returnn.frontend.device#

Device handling.

returnn.frontend.device.copy_to_device(x: Tensor, device: str | None = None) → Tensor[source]#

Copy tensor to device.

Parameters:

  • x – tensor

  • device –

Returns:

tensor on device

returnn.frontend.device.get_default_device() → str | None[source]#
Returns:

default device, where to put new tensors (via random number generators, constant, range_over_dim, etc)

returnn.frontend.device.set_default_device_ctx(device: str | None)[source]#
Parameters:

device – see get_default_device()