pytorch中nonzero()的用法

发布时间：2024-01-02 17:32:47

pytorch中nonzero()的用法

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

在这里插入图片描述

文章来源:https://blog.csdn.net/weixin_44928646/article/details/104629119
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
最新文章