pytorch 矩阵乘法

在 PyTorch 中，可以使用 `torch.matmul()` 函数来实现矩阵乘法。该函数可以接受两个张量作为输入，并返回它们的矩阵乘积。

例如，假设我们有两个 2x3 的张量 `A` 和 `B`，我们可以使用以下代码进行矩阵乘法：

import torch

A = torch.tensor([[1, 2, 3], [4, 5, 6]])
B = torch.tensor([[7, 8], [9, 10], [11, 12]])

C = torch.matmul(A, B)

print(C)

输出结果为：

tensor([[ 58,  64],
        [139, 154]])

在这个例子中，我们先定义了两个张量 `A` 和 `B`，然后使用 `torch.matmul()` 函数计算它们的矩阵乘积，并将结果存储在变量 `C` 中。最后，我们打印出 `C` 的值，得到了矩阵乘积的结果。