按element取自然对数:。
vec_ln(mask, dst, src, repeat_times, dst_rep_stride, src_rep_stride)
请参见参数说明。
dst与src的数据类型需要保持一致。
昇腾310 AI处理器,dst与src支持的数据类型为:Tensor(float16)
昇腾910 AI处理器,dst与src支持的数据类型为:Tensor(float16/float32)
昇腾310P AI处理器AI Core,dst与src支持的数据类型为:Tensor(float16/float32)
昇腾310P AI处理器Vector Core,dst与src支持的数据类型为:Tensor(float16/float32)
无。
昇腾310 AI处理器
昇腾910 AI处理器
昇腾310P AI处理器AI Core
昇腾310P AI处理器Vector Core
此样例是针对数据量较小、一次搬运就可以完成的场景,目的是让大家了解接口的功能,更复杂的数据量较大的样例可参见调用示例。
from tbe 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") # 将用户输入数据从gm搬运到ub 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) # 将计算结果从ub搬运到目标gm tik_instance.data_move(dst_gm, dst_ub, 0, 1, 8, 0, 0) tik_instance.BuildCCE(kernel_name="vec_ln", inputs=[src_gm], outputs=[dst_gm])
结果示例:
输入数据: [1, 2, 3, 4, ......, 128] 输出数据: [0, 0.6931, 1.0986, 1.3863, ......, 4.8520]