nvidia.dali.fn.squeeze#

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

移除作为 axes 或 axis_names 给定的维度。

移除会导致总体积发生变化的维度是错误的。

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

支持的后端

  • ‘cpu’

  • ‘gpu’

参数:

__data (TensorList) – 要挤压的数据

关键字参数:

  • axes (int 或 int 列表或 int 的 TensorList，可选，默认值 = []) –

    应移除的维度的索引。

    所有挤压的维度的大小都应为 1，除非张量的总体积在挤压前后均为 0。所有索引都必须在输入的有效维度范围内

  • axis_names (布局字符串, 可选，默认值 = ‘’) –

    应移除的布局列。

    所有挤压的维度的大小都应为 1，除非张量的总体积在挤压前后均为 0。所有布局名称都应存在于数据布局中。

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

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

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

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

另请参阅

在本页中