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) – 即使操作符的输出未使用，也阻止从图中删除操作符。

另请参阅

在本页中