Pytorch中的比较运算

 案例：

import torch

a = torch.rand(2,3)
b = torch.rand(2,3)
print(a)
print(b)

print(torch.eq(a, b))
print(torch.equal(a, b))

print(torch.ge(a, b))
print(torch.gt(a, b))
print(torch.le(a, b))
print(torch.lt(a, b))

# 排序
a = torch.tensor(data=[[3, 1, 4, 4], [32,3,4,3]])
print(torch.sort(a,dim=0, descending=True))


"""
tensor([[0.8548, 0.6416, 0.3903],
        [0.8100, 0.8877, 0.8933]])
        
tensor([[0.9877, 0.8922, 0.7746],
        [0.3026, 0.8608, 0.0538]])
        
tensor([[False, False, False],
        [False, False, False]])
        
False
tensor([[False, False, False],
        [ True,  True,  True]])
        
tensor([[False, False, False],
        [ True,  True,  True]])
        
tensor([[ True,  True,  True],
        [False, False, False]])
        
tensor([[ True,  True,  True],
        [False, False, False]])
        
torch.return_types.sort(
values=tensor([[32,  3,  4,  4],
        [ 3,  1,  4,  3]]),
indices=tensor([[1, 1, 0, 0],
        [0, 0, 1, 1]]))


"""