vrec
功能说明
对tensor中的每个元素做倒数运算 1 / x。
函数原型
vrec(raw_tensor, impl_mode="high_performance")
参数说明
- raw_tensor:输入tensor,tvm.tensor类型。
Atlas 200/300/500 推理产品:支持的数据类型有float16、float32
Atlas 训练系列产品:支持的数据类型有float16、float32
Atlas 推理系列产品(Ascend 310P处理器):支持的数据类型有float16、float32
Atlas 200I/500 A2推理产品:支持的数据类型有float16、float32
Atlas A2训练系列产品/Atlas 800I A2推理产品:支持的数据类型有float16、float32
- impl_mode:算子运行时选择精度优先还是性能优先模式,有以下两种取值:
- high_precision:表示精度优先,精度优先时,由于计算过程复杂,性能会有所下降。
- high_performance:表示性能优先,性能优先时,精度会有所下降。
默认值“high_performance”。
返回值
res_tensor:表示1 / raw_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) res = dsl.vrec(data)
父主题: Math计算接口