关于四维矩阵乘法 - pytorch吧

level 1

牛叔orz 楼主

如题，我现在需要一个shape=（9,10，435，200）的tensor和一个shape=（9,10，200，8）相乘获得一个shape=(9,10,435,8)的向量，但是用b
mm报错说要三维的张量。请问这样的四维相乘应该如何做到？

2020年02月24日 10点02分 1

level 1

牛叔orz 楼主

哦好像找到了用matmul，打扰了

2020年02月24日 10点02分 2