nvidia.dali.fn.copy#

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

创建输入张量的副本。

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

支持的后端
  • ‘cpu’

  • ‘gpu’

参数:

__input (TensorList) – 运算符的输入。

关键字参数:
  • bytes_per_sample_hint (int 或 int 列表, 可选, 默认值 = [0]) –

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

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

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