华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
问题如题，我尝试在核函数中传入一个标量参数，用于和向量做element-wise的运算。
尝试在工程json配置文件中加入attr参数
attr
"attr": [ { "name": "value", "param_type": "required", "type": "float" } ]
生成的工程在算子注册阶段确实将该参数注册进了算子
this->Attr("value").Float();
阅读官方文档后，发现这个attr貌似是Host端可用的参数，如果我想在核函数中使用标量参数该怎么处理呢？
我要发帖子
点赞
收藏
回复
分享
举报
已解决 已解决
问题如题，我尝试在核函数中传入一个标量参数，用于和向量做element-wise的运算。
尝试在工程json配置文件中加入
attr参数"attr": [ { "name": "value", "param_type": "required", "type": "float" } ]生成的工程在算子注册阶段确实将该参数注册进了算子
this->Attr("value").Float();阅读官方文档后，发现这个
attr貌似是Host端可用的参数，如果我想在核函数中使用标量参数该怎么处理呢？