vmaxs
功能说明
将raw_tensor中每个元素与标量scalar进行比较,结果取两者间的最大者。
函数原型
vmaxs(raw_tensor, scalar)
参数说明
- raw_tensor:输入tensor,tvm.tensor类型。
- scalar:raw_tensor中元素要比较的系数,标量类型。
- scalar数据类型与raw_tensor数据类型需要一致,如果不一致计算中会把scalar转换为raw_tensor对应数据类型。
Atlas 200/300/500 推理产品:支持的数据类型有float16、float32、int32
Atlas 训练系列产品:支持的数据类型有float16、float32、int32
Atlas 推理系列产品(Ascend 310P处理器):支持的数据类型有float16、int16、float32、int32
Atlas 200I/500 A2推理产品:支持的数据类型有float16、float32、int32
Atlas A2训练系列产品/Atlas 800I A2推理产品:支持的数据类型有float16、float32、int32
返回值
res_tensor:结果tensor,tvm.tensor类型。
约束说明
无。
支持的型号
Atlas 200/300/500 推理产品
Atlas 训练系列产品
Atlas 推理系列产品(Ascend 310P处理器)
Atlas 200I/500 A2推理产品
Atlas A2训练系列产品/Atlas 800I A2推理产品
调用示例
from tbe import tvm from tbe import dsl shape = (1024,1024) input_dtype = "float16" data = tvm.placeholder(shape, name="data", dtype=input_dtype) scalar = tvm.const(2, dtype =input_dtype) res = dsl.vmaxs(data, scalar)
父主题: Math计算接口