文档首页> Atlas 500应用> TE自定义算子开发指导> 自定义算子开发> 算子代码实现> 执行算子
更新时间：2021-03-18 GMT+08:00
查看PDF
分享

执行算子

自定义算子代码编写完成后，可以在算子的*.py代码结尾加入调用算子的语句，如下所示，可以参照算子运行验证构造输入数据，验证算子执行结果是否正确。

例如：

if __name__ == "__main__":
    reduction((2, 3, 4), "float16", 1, "SUM", coeff = 2,kernel_name = "Reduction")
父主题： 算子代码实现
分享：

    相关文档

    相关产品