【求助】LocalTensor如何获得矩阵的长度和宽度以及如何获得attr参数
收藏回复举报
【求助】LocalTensor如何获得矩阵的长度和宽度以及如何获得attr参数
t('forum.solved') 已解决
发表于2024-07-28 16:05:53
0 查看
如题,笔者在实现算子时发现文档中直接将算子看作一维的tensor,如果传入的是一个矩阵,该如何获取对应数据的长度和宽度,以及如何在kernel中获得attr参数呢?

我要发帖子