Shortcuts

mmengine.dist.cast_data_device

mmengine.dist.cast_data_device(data, device, out=None)[源代码]

Recursively convert Tensor in data to device.

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

参数:
  • 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.

返回:

data was casted to device.

返回类型:

Tensor or list or dict

Read the Docs v: latest
Versions
latest
stable
v0.10.3
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.