昇腾社区首页
中文
注册

multiply

函数功能

图像处理类算法,张量乘法multiply,支持dtype.float16、dtype.float32、dtype.uint8。

当前支持Atlas 推理系列产品Atlas 200I/500 A2 推理产品

  • 接口中的输入输出Tensor必须在Device侧。
  • 输入参数对应Tensor的形状(Shape)相等且不超过4维。
  • 当不传入scale时,输入参数的各数据类型需保持一致;若传入,支持不同输入数据类型,输出数据类型与精度高的输入Tensor保持一致。
  • 请注意处理数据类型越界问题。

函数原型

multiply(inputTensor1: Tensor, inputTensor2: Tensor, scale)

输入参数说明

参数名

类型

说明

inputTensor1

Tensor类

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

inputTensor2

Tensor类

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

scale

float

可选参数,表示结果被缩放的倍数。

返回参数说明

返回相乘后的Tensor类数据。