统计Tensor中False出现的次数
#mask 是一个这样的tensor: tensor([[False, False, False, ..., False, False, False]], device='cuda:0')
#shape: torch.Size([1, 1024])
print((mask==False).sum())
#tensor(1024, device='cuda:0')
统计Tensor中False出现的次数
#mask 是一个这样的tensor: tensor([[False, False, False, ..., False, False, False]], device='cuda:0')
#shape: torch.Size([1, 1024])
print((mask==False).sum())
#tensor(1024, device='cuda:0')