nvidia.dali.fn.cast

nvidia.dali.fn.cast(__input, /, *, bytes_per_sample_hint=[0], dtype, preserve=False, device=None, name=None)

将张量转换为不同的类型。

此操作器允许序列输入并支持体积数据。

支持的后端

• ‘cpu’

• ‘gpu’

参数:

__input (TensorList) – 操作器的输入。

关键字参数:

• bytes_per_sample_hint (int 或 int 列表, 可选, 默认 = [0]) –

  输出大小提示，以字节/样本为单位。

  如果指定，则操作器位于 GPU 或页锁定主机内存中的输出将被预分配以容纳此大小的样本批次。

• dtype (nvidia.dali.types.DALIDataType) – 输出数据类型。

• preserve (bool, 可选, 默认 = False) – 阻止操作器从图中移除，即使其输出未使用。