首页 > 其他分享> > pytorch----一个高维矩阵乘法(维度不同的矩阵乘法)引发的血案 pytorch----一个高维矩阵乘法(维度不同的矩阵乘法)引发的血案 2020-12-11 13:01:07 作者:互联网 背景: 要实现这个矩阵乘法 方案一: 自己低级的实现:每一个维度做for循环 方案二: 师兄的方案, 大佬的实现 方案三: 和义朋兄讨论之后, 实现的方案: torch.matmul()实现 即, matmul直接可以实现维度不同的矩阵乘法 标签:方案,实现,矩阵,----,matmul,维度,乘法 来源: https://blog.csdn.net/weiwei935707936/article/details/111032629