pytorch----一个高维矩阵乘法(维度不同的矩阵乘法)引发的血案2024-03-24 17:33:58背景: 要实现这个矩阵乘法 方案一: 自己低级的实现:每一个维度做for循环 方案二: 师兄的方案, 大佬的实现 方案三: 和义朋兄讨论之后, 实现的方案: torch.matmul()实现 即, matmul直接可以实现维度不同的矩阵乘法 上一篇:AI推理与Compiler下一篇:TensorFlow常用函数