PyTorch中tensor.eq()和tensor.lt()

在PyTorch中Tensor的查找和筛选例子

>>> x = torch.arange(5)  
>>> x
tensor([0, 1, 2, 3, 4])
>>> torch.gt(x,1) # 大于
tensor([0, 0, 1, 1, 1], dtype=torch.uint8)
>>> x>1   # 大于
tensor([0, 0, 1, 1, 1], dtype=torch.uint8)
>>> torch.ne(x,1) # 不等于
tensor([1, 0, 1, 1, 1], dtype=torch.uint8)
>>> x!=1  # 不等于
tensor([1, 0, 1, 1, 1], dtype=torch.uint8)
>>> torch.lt(x,3) # 小于
tensor([1, 1, 1, 0, 0], dtype=torch.uint8)
>>> x<3   # 小于
tensor([1, 1, 1, 0, 0], dtype=torch.uint8)
>>> torch.eq(x,3) # 等于
tensor([0, 0, 0, 1, 0], dtype=torch.uint8)
>>> x==3  # 等于
tensor([0, 0, 0, 1, 0], dtype=torch.uint8)

>>> a= torch.arange(1,5)
>>> print(a)
tensor([1, 2, 3, 4])
>>> a.eq(1)
tensor([ True, False, False, False])
>>> a.lt(1)
tensor([False, False, False, False])
>>> a.lt(2).float()
tensor([1., 0., 0., 0.])
>>> a.eq(2).float()
tensor([0., 1., 0., 0.])

PyTorch中的sigmoid_函数

>>> a= torch.ones(5).float()
>>> a
tensor([1., 1., 1., 1., 1.])
>>> a.sigmoid_()
tensor([0.7311, 0.7311, 0.7311, 0.7311, 0.7311])