Shortcuts

mmengine.dist.cast_data_device

mmengine.dist.cast_data_device(data, device, out=None)[source]

Recursively convert Tensor in data to device.

If data has already on the device, it will not be casted again.

Parameters
  • data (Tensor or list or dict) – Inputs to be casted.

  • device (torch.device) – Destination device type.

  • out (Tensor or list or dict, optional) – If out is specified, its value will be equal to data. Defaults to None.

Returns

data was casted to device.

Return type

Tensor or list or dict

Read the Docs v: stable
Versions
latest
stable
v0.10.2
v0.10.1
v0.10.0
v0.9.1
v0.9.0
v0.8.5
v0.8.4
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.