建议使用以下浏览器,以获得最佳体验。 IE 9.0+以上版本 Chrome 31+ 谷歌浏览器 Firefox 30+ 火狐浏览器
请选择 进入手机版 | 继续访问电脑版
设置昵称

在此一键设置昵称,即可参与社区互动!

确定
我再想想
选择版块
092713ie8gnff9tx2p8m72.jpg MindSpore!这款刚刚开源的深度学习框架我爱了 知乎大咖入驻昇腾社区 B站UP主入驻昇腾社区

fengkaige

发帖: 3粉丝: 0

级别 : 新手上路

发消息 + 关注

发表于2020-5-28 14:58:12 217 3
直达本楼层的链接
楼主
显示全部楼层
[问题求助] 关于想直接调用TE API的问题

关于TE API的直接调用问题

像下面这个TE里面的API : "te.lang.cce.vmuls(raw_tensor,scalar)"

image.png

想直接调用这个API,然后传数据,进行乘法运算,代码如下:

#-*- coding:utf-8 -*-
import te.lang.cce 
from te import tvm 
import topi
import numpy as np
# 使用算子
# a = tvm.nd.array(np.random.uniform(size=n).astype(dtype))
data1 = tvm.nd.array(np.array([[60000]]).astype(dtype))
data2 = tvm.nd.array(np.array([60000]).astype(dtype))
# data1 = tvm.tensor.Tensor(np.array([[60000]]).astype(dtype))
print("=======",data1)
scalar = tvm.const(60000dtype =dtype) 
res1 = te.lang.cce.vmuls(data1,scalar)
print("*****",res1)


但运行一直说传入的不是个tensor,使用tvm传tensor的方式也不行,不知道这个API接口可以直接传数据计算吗?可以的话怎么传可否给个示例?因为看这些示例都是在自定义算子的时候用的占位符之类的传入数据的,没有直接传数据的。image.png

举报
分享

分享文章到朋友圈

分享文章到微博

xiaoyifan6

发帖: 1粉丝: 4

级别 : 版主

发消息 + 关注

发表于2020-5-28 16:13:15
直达本楼层的链接
沙发
显示全部楼层

好的,问题已经收录,我们会尽快给您答复。

点赞 评论 引用 举报

A文弱书生

发帖: 0粉丝: 0

级别 : 版主

发消息 + 关注

发表于2020-6-3 14:51:46
直达本楼层的链接
板凳
显示全部楼层

直接输入数据应该是支持的,dsl的输入要求是tensor dict吧, 你这里只是个array

算子开发可参考https://support.huawei.com/enterprise/zh/doc/EDOC1100115566/6e2c7d81

算子开发样例:https://gitee.com/HuaweiAtlas/AtkOps

点赞 评论 引用 举报

A文弱书生

发帖: 0粉丝: 0

级别 : 版主

发消息 + 关注

发表于2020-6-3 15:55:04
直达本楼层的链接
地板
显示全部楼层

你下面注释掉那个,你构造个完整的应该就行了,要有shape,dtype啥的

点赞 评论 引用 举报

游客

富文本
Markdown
您需要登录后才可以回帖 登录 | 立即注册