你的浏览器版本过低,可能导致网站不能正常访问!
为了你能正常使用网站功能,请使用这些浏览器。

Using torch.matmul in pytorch prompts a mismatch of multiplication inputs

[复制链接]
捻墨玗埃 提问时间:2025-3-1 13:46 / 未解决

When penalizing a 3D matrix in pytorch using torch.matmul, it is multiplying the last two latitudes, and when converting to onnx to put into the st platform, the realistic multiplication inputs do not match.

How should this be solved?

p1.jpg

赞 收藏 评论2 发布时间:2025-3-1 13:46

举报

2个回答
捻墨玗埃 回答时间:2025-3-1 14:24:25

p1.jpg

yijun.deng 回答时间:2025-3-5 16:45:01

Hello,

The tool has treated the dimensions to be H, W, C, but actually your first dimension should be batch, I suggest you to use 2D Matmul.

Best Regards,

Yijun

所属标签

相似问题

官网相关资源

关于
我们是谁
投资者关系
意法半导体可持续发展举措
创新与技术
意法半导体官网
联系我们
联系ST分支机构
寻找销售人员和分销渠道
社区
媒体中心
活动与培训
隐私策略
隐私策略
Cookies管理
行使您的权利
官方最新发布
人形机器人运动控制、感知与智能配电
半导体创新技术与应用方向
EE架构与软件定义汽车
12V/48V 汽车智能配电(SPD)
区域控制单元(ZCU)与分区架构
关注我们
st-img 微信公众号
st-img 手机版