关于全连接层linear的层维度匹配问题 All you have to care isb=cand you are done:m1: [a x b], m2: [c x d]m1is[a x b]which is[batch size x in features]m2is[c x d]which is[in features x out features] pytorch 2023-08-10 44 点赞 0 评论 66 浏览
