divide
函数功能
图像处理类算法,张量除法divide,支持dtype.float16、dtype.float32、dtype.uint8。
当前支持Atlas 推理系列产品和Atlas 200I/500 A2 推理产品。
- 接口中的输入输出Tensor必须在Device侧。
- 输入参数对应Tensor的形状(Shape)相等且不超过4维。
- scale参数仅支持在Atlas 推理系列产品下传入。支持不同输入数据类型,输出数据类型与精度高的输入Tensor保持一致。
- 请注意处理数据类型越界问题。
函数原型
divide(inputTensor1: Tensor, inputTensor2: Tensor, scale)
输入参数说明
参数名 |
类型 |
说明 |
---|---|---|
inputTensor1 |
Tensor类 |
被除数,输入张量,支持dtype.float16、dtype.float32、dtype.uint8类型输入。 |
inputTensor2 |
Tensor类 |
除数,输入张量,支持dtype.float16、dtype.float32、dtype.uint8类型输入。 |
scale |
float |
可选参数,表示被除数被缩放的倍数(该参数仅支持在Atlas 推理系列产品下使用)。 |
返回参数说明
返回相除后的Tensor类数据。
父主题: TensorOperations