PyTorch torch.linalg.matmul 函数
Pytorch torch 参考手册torch.linalg.matmul 是 PyTorch 线性代数模块中用于执行矩阵乘法的函数。它是 torch.matmul 的别名，提供了更统一的线性代数接口。
函数定义
torch.linalg.matmul(input, other, out=None)
参数:
input(Tensor): 第一个输入张量。other(Tensor): 第二个输入张量。out(Tensor, 可选): 输出张量。
返回值:
torch.Tensor: 返回矩阵乘法的结果。
使用示例
实例
import torch
# 创建两个矩阵
A = torch.randn(3, 4)
B = torch.randn(4, 5)
# 矩阵乘法
C = torch.linalg.matmul(A, B)
print("A 的形状:", A.shape)
print("B 的形状:", B.shape)
print("C 的形状:", C.shape)
# 创建两个矩阵
A = torch.randn(3, 4)
B = torch.randn(4, 5)
# 矩阵乘法
C = torch.linalg.matmul(A, B)
print("A 的形状:", A.shape)
print("B 的形状:", B.shape)
print("C 的形状:", C.shape)
输出结果为：
A 的形状: torch.Size([3, 4]) B 的形状: torch.Size([4, 5]) C 的形状: torch.Size([3, 5])