nvidia.dali.fn.stack#

nvidia.dali.fn.stack(__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) –

    输出张量中输入堆叠所在的轴。

    该轴插入到输入中相应轴的前面。值 0 表示堆叠整个张量。指定等于输入维数的 axis 会导致输入中的值被交错）。

    接受的范围是 [-ndim, ndim]。负索引从末尾计数。

  • axis_name (str, 可选) –

    要插入的新轴的名称。

    一个字符的字符串，将在输出布局中表示新轴。输出布局将通过将该字符插入到由 axis 指示的位置的输入布局中来构建。例如，指定 axis = 0 和 axis_name = "C" 且输入布局为 “HW” 将产生输出布局 “CHW”

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

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

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

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

另请参阅

在本页中