文档首页>
Atlas 200 DK 开发者套件(1.0.7.alpha)>
TBE自定义算子开发指南>
接口参考>
单目(Gather模式)>
TIK API>
矢量计算>
算子代码实现接口>
vec_ln
更新时间:2021-03-18 GMT+08:00
vec_ln
功能说明
按element取自然对数:。
对于昇腾310 AI处理器,使用该接口处理float16类型的数据时,如果输入数据在(0, 5/3)区间,计算精度不满足双千分之一,如果对精度有更高要求,建议使用vec_ln_high_preci接口。
函数原型
vec_ln(mask, dst, src, repeat_times, dst_rep_stride, src_rep_stride)
返回值
无。
注意事项
- 如果src的数值为非正数,可能会产生未知结果。
- 其它注意事项请参考注意事项。
调用示例
from te import tik tik_instance = tik.Tik() # 申请tensor src_gm = tik_instance.Tensor("float16", (128,), tik.scope_gm, "src_gm") src_ub = tik_instance.Tensor("float16", (128,), name="src_ub", scope=tik.scope_ubuf) dst_ub = tik_instance.Tensor("float16", (128,), name="dst_ub", scope=tik.scope_ubuf) dst_gm = tik_instance.Tensor("float16", (128,), tik.scope_gm, "dst_gm") # 拷贝用户输入数据到src ubuf tik_instance.data_move(src_ub, src_gm, 0, 1, 8, 0, 0) tik_instance.vec_ln(128, dst_ub, src_ub, 1, 8, 8) # 将计算结果拷贝到目标gm tik_instance.data_move(dst_gm, dst_ub, 0, 1, 8, 0, 0) tik_instance.BuildCCE("v100_mini_vec_ln_test", [src_gm], [dst_gm]) 输入数据: [1, 2, 3, 4, ......, 128] 输出数据: [0, 0.6931, 1.0986, 1.3863, ......, 4.8520]
父主题: 单目(Gather模式)
