当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python mxnet.ndarray.op.squeeze用法及代码示例


用法:

mxnet.ndarray.op.squeeze(data=None, axis=_Null, out=None, name=None, **kwargs)

参数

  • data(NDArray) - 要压缩的数据
  • axis(Shape or None, optional, default=None) - 选择形状中一维条目的子集。如果选择了形状条目大于 1 的轴,则会引发错误。
  • out(NDArray, optional) - 输出 NDArray 来保存结果。

返回

out- 此函数的输出。

返回类型

NDArray 或 NDArray 列表

从数组的形状中删除一维条目。在大多数情况下,将输出张量形状定义为 numpy.squeeze 的行为相同。有关例外情况,请参阅以下注释。例子:

data = [[[0], [1], [2]]]
squeeze(data) = [0, 1, 2]
squeeze(data, axis=0) = [[0], [1], [2]]
squeeze(data, axis=2) = [[0, 1, 2]]
squeeze(data, axis=(0, 2)) = [0, 1, 2]

注意

此运算符的输出将保持至少一个维度不被删除。例如,squeeze([[[4]]]) = [4],而在numpy.squeeze 中,输出将变为标量。

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.op.squeeze。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。