星驰编程网

import torch

# 创建两个2x2矩阵
matrix1 = torch.tensor([[1, 2], [3, 4]])
matrix2 = torch.tensor([[5, 6], [7, 8]])

# 计算矩阵乘积
result = torch.matmul(matrix1, matrix2)  # 或者使用 matrix1 @ matrix2

print("矩阵1:")
print(matrix1)
print("\n矩阵2:")
print(matrix2)
print("\n矩阵乘积:")
print(result)

这段代码使用PyTorch的tensor创建了两个2x2矩阵，并通过torch.matmul()函数计算它们的乘积。你也可以使用@运算符进行矩阵乘法。输出将显示原始矩阵和它们的乘积结果。

矩阵1:
tensor([[1, 2],
        [3, 4]])

矩阵2:
tensor([[5, 6],
        [7, 8]])

矩阵乘积:
tensor([[19, 22],
        [43, 50]])