华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
昇腾社区首页
昇腾论坛
0
/100
支持
消息
我的论坛主页
论坛首页
/
关于扩大算子输入shape的方法
点赞
0
收藏
0
回复
0
分享
举报
只看楼主
关于扩大算子输入shape的方法
已解决
发表于2024-04-30 10:10:07
0
查看
老师您好,我现在实现了一个可以处理矩阵乘法算子输入规模为2个32*32的half类型,输出规模为1个32*32的float类型的算子,现在想对其输入输出规模进行扩大,例如扩大到10240*10240个half类型的输入,1个10240*10240个float类型的输出,但是直接扩张的话存储不够,有什么办法能分块输入吗,或者有类似的例子可以学习以下处理方法吗
匿名回复
发表
我要发帖子
我们使用cookie来确保您的高速浏览体验。继续浏览本站,即表示您同意我们使用cookie。
查看详情