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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/jacke121/article/details/82818140

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))

阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页