pytorch 几个不同获得索引的方法

pytorch 几个不同获得索引的方法

https://blog.csdn.net/luolinll1212/article/details/82224491

2018年08月30日 17:08:36 荷塘翠柳 阅读数：46 标签： 深度学习pytorch 更多

个人分类： pytorch深度学习

                                  运算      函数

                                  大于     torch.gt

                                  小于     torch.lt

                                  等于     torch.eq

                                  非零     torch.nonzero

                                  非         torch.ne

1. import torch

2.  

3. x = torch.arange(5)

4. print(x)

5. mask = torch.gt(x,1) # 大于

6. print(mask)

7. print(x[mask])

8.  

9. x = torch.arange(5)

10. print(x)

11. mask = torch.lt(x,3) # 小于

12. print(mask)

13. print(x[mask])

14.  

15. x = torch.arange(5)

16. print(x)

17. mask = torch.eq(x,3) # 等于

18. print(mask)

19. print(x[mask])

20.  

21. x = torch.Tensor([1,2,1,0,0])

22. mask = torch.ne(x,1) # 非，一个数

23. print(mask)

24. print(x[mask])

25.  

26. a = torch.Tensor([[0.6, 0.0, 0.0, 0.0],[0.0, 0.4, 0.0, 0.0],[0.0, 0.0, 1.2, 0.0],[0.0, 0.0, 0.0,-0.4]])

27. mask = torch.nonzero(a) # 非零

28. print(mask)

29. print(torch.numel(mask))

30. print(torch.numel(a))

31. # print(a[mask])

32. print(torch.numel(mask)/torch.numel(a))