关于扩大算子输入shape的方法
收藏回复举报
关于扩大算子输入shape的方法
t('forum.solved') 已解决
发表于2024-04-30 10:10:07
0 查看
老师您好,我现在实现了一个可以处理矩阵乘法算子输入规模为2个32*32的half类型,输出规模为1个32*32的float类型的算子,现在想对其输入输出规模进行扩大,例如扩大到10240*10240个half类型的输入,1个10240*10240个float类型的输出,但是直接扩张的话存储不够,有什么办法能分块输入吗,或者有类似的例子可以学习以下处理方法吗

我要发帖子