其他分享
首页 > 其他分享> > pytorch----一个高维矩阵乘法(维度不同的矩阵乘法)引发的血案

pytorch----一个高维矩阵乘法(维度不同的矩阵乘法)引发的血案

作者:互联网

背景:  要实现这个矩阵乘法

 

 

方案一: 自己低级的实现:每一个维度做for循环

 

方案二: 师兄的方案, 大佬的实现

 

 

方案三: 和义朋兄讨论之后, 实现的方案: torch.matmul()实现

标签:方案,实现,矩阵,----,matmul,维度,乘法
来源: https://blog.csdn.net/weiwei935707936/article/details/111032629