nvidia.dali.experimental.dynamic.cast

nvidia.dali.experimental.dynamic.cast(input, /, *, batch_size=None, device=None, dtype)

Cast a tensor to a different type.

This operator allows sequence inputs and supports volumetric data.

Supported backends

• ‘cpu’

• ‘gpu’

Parameters:

input (Tensor/Batch) – Input to the operator.

Keyword Arguments:

• dtype (nvidia.dali.types.DALIDataType) – Output data type.

• batch_size (int, optional) – The batch size to broadcast input tensors to. Ignored for batch inputs.

• device (device-like, optional) – The device to use for the operation. Must not conflict with the device of the inputs.