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.