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

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

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

 

 

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

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

 

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

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

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

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

 

 

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

  • 即, matmul直接可以实现维度不同的矩阵乘法

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

上一篇:AI推理与Compiler


下一篇:TensorFlow常用函数