华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
pytorch模型迁移Mindspore过程中,原pytorch代码的矩阵相乘操作没有问题,而在mindspore中则参数类型错误
TypeError: For matmul op, inputs must be all tensors.
其中,ent_ent为SparseTensor,he_emb为Tensor
torch.matmul(ent_ent, he_emb) mindspore.ops.matmul(ent_ent, he_emb)
想要运行是必须两个都是Tensor吗
我要发帖子
pytorch模型迁移Mindspore过程中,原pytorch代码的矩阵相乘操作没有问题,而在mindspore中则参数类型错误
TypeError: For matmul op, inputs must be all tensors.
其中,ent_ent为SparseTensor,he_emb为Tensor
想要运行是必须两个都是Tensor吗