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) – 阻止算子从图中移除，即使其输出未使用。

另请参阅

本页内容