Tensor的比较运算

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))
print(torch.ne(a, b))

a = torch.tensor([[1, 4, 4, 3, 5],[2, 3, 1, 3, 5]])
print(torch.sort(a))
print(torch.sort(a, descending=True))
print(torch.sort(a, dim= 0, descending=True))
print(torch.sort(a, dim= 1, descending=True))

topk运算

a= torch.tensor([[2, 4, 3, 1, 5],
                 [2, 3, 5, 1, 4]])
print(a.shape)

print(torch.topk(a, k = 1 , dim=0))

a = torch.rand(2, 3)
print(a)
print(a/0)
print(torch.isfinite(a))
print(torch.isfinite(a/0))
print(torch.isinf(a/0))
print(torch.isnan(a))

import numpy as np
a = torch.tensor([1, 2, np.nan])
print(torch.isnan(a))