nvidia.dali.fn.cast_like

nvidia.dali.fn.cast_like(__input_0, __input_1, /, *, bytes_per_sample_hint=[0], preserve=False, device=None, name=None)

将第一个张量转换为第二个张量的类型。

此运算符允许序列输入并支持体积数据。

支持的后端

• ‘cpu’

• ‘gpu’

参数:

• __input_0 (TensorList) – 运算符的输入。

• __input_1 (TensorList) – 运算符的输入。

关键字参数:

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

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

  如果指定，则运算符位于 GPU 或页锁定主机内存中的输出将被预先分配，以适应此大小的样本批次。

• preserve (bool, 可选, 默认 = False) – 即使运算符的输出未使用，也阻止将其从图中删除。