当前位置：网站首页>pytorch，nonzero 实例 使用

pytorch，nonzero 实例 使用

2022-07-19 12:33:00 【51CTO】 


      
      

       
       import torch
       
       
input_tensor = torch.tensor([1,2,3,4,5])
       
       
mask = input_tensor>3
       
       
print(mask)
       
       
indexes = mask.nonzero().squeeze()
       
       
print(indexes)
      
      

      
      
    
       
       
  • 1.
    • 
       
       
  • 2.
    • 
       
       
  • 3.
    • 
       
       
  • 4.
    • 
       
       
  • 5.
    • 
       
       
  • 6.
    • 
      
      
      
      

       
       tensor([0, 0, 0, 1, 1], dtype=torch.uint8)
       
       
tensor([3, 4])
      
      

      
      
    
       
       
  • 1.
    • 
       
       
  • 2.
    •


原网站

版权声明
本文为[51CTO]所创，转载请带上原文链接，感谢
https://blog.51cto.com/guotong1988/5486080

边栏推荐

猜你喜欢

随机推荐