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 或 list of int 或 TensorList of int, optional, default = []) –

    应移除的维度的索引。

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

  • axis_names (layout str, optional, default = ‘’) –

    应移除的布局列。

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

  • bytes_per_sample_hint (int 或 list of int, optional, default = [0]) –

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

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

  • preserve (bool, optional, default = False) – 阻止运算符即使在其输出未使用时也从图中移除。

另请参阅

在本页中