举例说明torch.matmul可以用于多个矩阵的乘法

假设我们有三个矩阵A、B和C，形状分别为(2, 3)、(3, 4)和(4, 5)，那么可以使用torch.matmul计算它们的乘积：

import torch

A = torch.randn(2, 3)
B = torch.randn(3, 4)
C = torch.randn(4, 5)

result = torch.matmul(torch.matmul(A, B), C)

print(result.shape)

输出结果为torch.Size([2, 5])，表示结果矩阵的形状为(2, 5)。

在上述代码中，首先使用torch.matmul计算A和B的乘积，得到一个形状为(2, 4)的矩阵。然后再将这个结果与矩阵C相乘，得到最终的结果矩阵。可以看到，torch.matmul可以方便地进行多个矩阵的乘法运算。