nvidia.dali.fn.expand_dims#

nvidia.dali.fn.expand_dims(__data, /, *, axes, bytes_per_sample_hint=[0], new_axis_names='', preserve=False, device=None, name=None)#

向数据形状插入维度为 1 的新维度。

新维度插入的位置由 axes 指定。

如果提供了 new_axis_names，则新维度名称将插入到数据布局中，位置由 axes 指定。如果未提供 new_axis_names，则输出数据布局将为空。”

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

支持的后端

  • ‘cpu’

  • ‘gpu’

参数:

__data (TensorList) – 要扩展的数据

关键字参数:

  • axes (int 或 int 列表或 int 的 TensorList) – 插入新维度的索引。

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

    每个样本的输出大小提示（以字节为单位）。

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

  • new_axis_names (layout str, 可选, 默认值 = ‘’) –

    数据布局中新维度的名称。

    new_axis_names 的长度必须与 axes 的长度匹配。 如果未提供参数，则布局将被清除。

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

另请参阅

在本页中