python中tensor小妙用-----比较符号

在pytorch中tensor张量是可以和一个数或者另一个张量进行比较的，但是张量和张量之间进行比较时，需要做到两个张量的纬度相同。比较符号有（>< =等），最后会返回一个与张量纬度大小的张量，张量里面对应的值是一个逻辑数据类型

 1  a = torch.Tensor([[1,2],[3,4]])
 2  b = a > 1
 3 
 4  c =  torch.Tensor([[1,1],[4,5]])
 5  d = a > c
 6 
 7  print(b)
 8  print(d)
 9 
10 
11 
12 
13 tensor([[False,  True],
14         [ True,  True]])
15 
16 tensor([[False,  True],
17         [ False,  False]])

可以看到返回的值是一个逻辑数据类型的张量，同时可以根据这个张量对a内的值进行索引

 1 f = a[b] 2 print(f) 3 4 5 tensor([2., 3., 4.]) 

posted @ 2022-03-14 15:45  maple_hx  阅读(376)  评论(0)    收藏  举报