昇腾社区首页
中文
注册

ReduceMin

功能

沿着指定的轴计算输入Tensor元素的最小值。如果 keepdims 等于1,那么生成的Tensor与输入Tensor具有相同的维数。如果 keepdims 等于 0,那么生成的Tensor会去除被缩减的维度。

输入

data:输入Tensor,数据类型:uint8、int32、float16、float。数据格式支持ND。

输出

reduced data:输出Tensor,数据类型同输入data一致。

属性

  • keepdims:数据类型为int;是否保留缩减的维度;默认为1(保留)。
  • axes(可选):数据类型为Int64;指定计算轴;取值范围:[-r, r-1],r是输入数据data的维数。

限制与约束

axes为空时,当前输出不做全维度规约。此时建议修改onnx算子的axes为所有轴;或者在torch导出onnx图前,在torch模型中使用amax来对所有轴规约(例如,x.amax(dim=[0, 1, 2]))。

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18