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(device: str | None)[source]¶
- Parameters:
device – see
get_default_device()
- returnn.frontend.device.set_default_device_ctx(device: str | None)[source]¶
- Parameters:
device – see
get_default_device()