nonzero():返回一个包含输入Input中非0元素索引的张量(以张量的形式返回非0索引)
一维中:
import torch
a =torch.Tensor([1,2,4,4,5])
print(torch.nonzero(a<4))
a<4:输出的是布尔值索引(掩码)[1,1,0,0,0]
print(torch.nonzero(a<4)):输出布尔值索引非0的位置
二维
import torch
box=[]
cls=torch.Tensor([[0.6,0.4,0.8,0.7]])
print(cls>0.6)
idxs=torch.nonzero(torch.gt(cls,0.6))
print(idxs)
import torch
box=[]
cls=torch.Tensor([[0.6],[0.4],[0.8],[0.7]])
print(cls>0.6)
idxs=torch.nonzero(torch.gt(cls,0.6))
import torch
import numpy as np
a = torch.Tensor([[1,2],[3,4],[5,6],[7,8],[9,10]])
b = torch.Tensor([1,2,3,4,5])
print(a>3)
# print(a[a>3])
print(torch.nonzero(a>3))