Pytorch torch.nonzero()的简单用法

官方文档：https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch/
顾名思义，返回非零元素的索引：
先从一维的看起：

import torchx = torch.Tensor([0, 1, 2, 3, 0, 5])
y = torch.nonzero(x)
print(y)
print(y.size())

输出：

tensor([[1],[2],[3],[5]])
torch.Size([4, 1])

注意这里，首先输出y的维度为4×1，4表示非零元素的数量，1表示我们的输入维度。
再看一个二维的例子：

import torchx = torch.Tensor([[0, 1, 2, 3, 0, 5],[0, 0, 0, 0, 0, 10]])
y = torch.nonzero(x)
print(y)
print(y.size())
for i in y:print(i[0].numpy(), i[1].numpy())

输出：

tensor([[0, 1],[0, 2],[0, 3],[0, 5],[1, 5]])
torch.Size([5, 2])
0 1
0 2
0 3
0 5
1 5

输出y的维度为5×2，5表示非零元素的数量，2表示我们的输入维度。输出的每一行提供了相应非零元素的索引。

Pytorch torch.nonzero()的简单用法相关推荐

Pytorch torch.rsqrt()的简单用法
官方文档:https://pytorch.org/docs/master/generated/torch.rsqrt.html 简单来说就是对每个元素取平方根后再取倒数,即: out i=1input ...
Pytorch torch.topk()的简单用法
官方文档:https://pytorch.org/docs/stable/generated/torch.topk.html?highlight=topk#torch.topk 由于numpy本身是没 ...
Pytorch torch.mean()的简单用法
简单来说就是求平均数. 比如以下的三种简单情况: import torchx1 = torch.Tensor([1, 2, 3, 4]) x2 = torch.Tensor([[1],[2],[3], ...
Pytorch torch.manual_seed()的简单用法
简单来说就是指定随机数种子(种子可以是个任意int值),用来生成伪随机数: import torchtorch.manual_seed(1) print(torch.rand(1)) print(to ...
Pytorch nn.DataParallel()的简单用法
简单来说就是使用单机多卡进行训练. 一般来说我们看到的代码是这样的: net = XXXNet() net = nn.DataParallel(net) 这样就可以让模型在全部GPU上训练. 方法定义 ...
Pytorch transforms.Resize()的简单用法
简单来说就是调整PILImage对象的尺寸,注意不能是用io.imread或者cv2.imread读取的图片,这两种方法得到的是ndarray. 将图片短边缩放至x,长宽比保持不变: transfor ...
PyTorch：torch.nonzero——非零元素的定位
PyTorch学习笔记:torch.nonzero--非零元素的定位 torch.nonzero(input, *, out=None, as_tuple=False) → LongTensor or ...
【Torch API】torch.nonzero用法详解
torch.nonzero(input, *, out=None, as_tuple=False) → LongTensor or tuple of LongTensors 功能:用于输出数组的非零值 ...
pytorch tensor查找0_Pytorch简单教程
2019 年,ML 框架之争只剩两个实力玩家:PyTorch 和 TensorFlow.研究者大批涌向 PyTorch,而业界的首选仍然是 TensorFlow. Pytorch和TensorFlow ...

Pytorch torch.nonzero()的简单用法

Pytorch torch.nonzero()的简单用法相关推荐

最新文章

热门文章