昇腾社区首页
中文
注册

max_operator

函数功能

图像处理类算法,张量取较大值计算max_operator,对两个输入Tensor按元素比较并取较大值,支持dtype.float16、dtype.float32、dtype.uint8类型。

  • 当前支持Atlas 推理系列产品Atlas 200I/500 A2 推理产品
  • 接口中的输入Tensor必须在Device侧。
  • 输入参数对应Tensor的类型和形状(Shape)一致且形状不超过4维。

函数原型

max_operator(inputTensor1: Tensor, inputTensor2: Tensor,)

输入参数说明

参数名

类型

说明

inputTensor1

Tensor类

输入张量,支持dtype.float16、dtype.float32、dtype.uint8类型输入。

inputTensor2

Tensor类

输入张量,支持dtype.float16、dtype.float32、dtype.uint8类型输入。

返回参数说明

返回进行逐项取较大值后的Tensor类数据。