nvidia.dali.fn.cat#

nvidia.dali.fn.cat(__input_0, /, *__input_, axis=0, axis_name=None, bytes_per_sample_hint=[0], preserve=False, device=None, name=None)#

沿现有轴连接输入张量。

除了连接轴之外，输入形状在所有维度上必须匹配。

支持的后端

  • ‘cpu’

  • ‘gpu’

参数:

  • __input_0 (TensorList) – 操作符的输入。

  • __input_[1..998] (TensorList, 可选) – 此函数最多接受 998 个可选的位置输入

关键字参数:

  • axis (int, 可选, 默认 = 0) –

    输入张量沿此轴连接。

    接受的范围是 [-ndim, ndim-1]。负索引从后面计数。

  • axis_name (str, 可选) –

    张量连接轴的名称。

    此参数与 axis 互斥。此参数要求至少一个输入具有非空布局，并且所有非空输入布局都匹配。

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

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

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

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

另请参阅

在本页中