如何用pytorch写数字矩阵

可以用 PyTorch 的 tensor 类型来表示和处理数字矩阵，例如：

import torch

# 创建一个 3x3 的矩阵
matrix = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 打印矩阵
print(matrix)

输出：

tensor([[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9]])

可以使用 PyTorch 的 tensor 类型支持的函数来对数字矩阵进行各种操作，例如：

# 矩阵转置
transpose_matrix = matrix.transpose(0, 1)
print(transpose_matrix)

# 矩阵相乘
mul_matrix = torch.matmul(matrix, transpose_matrix)
print(mul_matrix)

# 矩阵元素求和
sum_matrix = torch.sum(matrix)
print(sum_matrix)

输出：

tensor([[1, 4, 7],
        [2, 5, 8],
        [3, 6, 9]])
tensor([[14, 32, 50],
        [32, 77,122],
        [50,122,194]])
tensor(45)

当然，这只是一个简单的示例，PyTorch 有着更多丰富的操作和功能，可以根据实际需求进行使用。