pytorch四维矩阵如何相乘2维矩阵

可以使用torch.matmul()函数来实现。需要保证矩阵维度的匹配关系，即第一个矩阵的行数等于第二个矩阵的列数。具体代码实现如下：

import torch

tensor_4d = torch.randn(2, 3, 4, 5)
tensor_2d = torch.randn(5, 2)

result = torch.matmul(tensor_4d, tensor_2d)

print(result.size()) # 输出为 (2, 3, 4, 2)