pytorch判断矩阵对称

在PyTorch中，你可以通过直接比较矩阵与其转置来判断矩阵是否是对称的。如果你有一个二维张量`matrix`，可以使用以下步骤：

import torch

def is_symmetric(matrix):
    # 获取矩阵的转置
    transposed_matrix = matrix.t()
    
    # 检查转置后的矩阵是否等于原始矩阵，忽略元素顺序带来的浮点数差异
    epsilon = 1e-6  # 可调整的小数值，用于处理浮点数精度问题
    return torch.allclose(matrix, transposed_matrix, rtol=epsilon, atol=epsilon)

这里`torch.allclose()`函数会检查两个张量是否几乎相等，`rtol` (relative tolerance) 和 `atol` (absolute tolerance) 控制了误差的范围。

例如，你可以像这样使用这个函数：

matrix = torch.tensor([[1, 2], [2, 1]])
if is_symmetric(matrix):
    print("矩阵是对称的")
else:
    print("矩阵不是对称的")