在PyTorch张量中查找等于零的元素的索引
，可以使用torch.nonzero()函数来实现。该函数返回一个包含所有非零元素索引的张量。
具体步骤如下：
- 导入PyTorch库：
import torch - 创建一个PyTorch张量：
tensor = torch.tensor([1, 0, 2, 0, 3, 0]) - 使用
torch.nonzero()函数查找等于零的元素的索引：indices = torch.nonzero(tensor == 0) - 打印结果：
print(indices)
这样就可以得到一个包含等于零元素索引的张量。
推荐的腾讯云相关产品是腾讯云AI智能图像处理服务，该服务提供了丰富的图像处理能力，包括图像识别、图像分析、图像增强等功能，可以与PyTorch等深度学习框架结合使用，实现更多复杂的图像处理任务。产品介绍链接地址：https://cloud.tencent.com/product/aiip
注意：以上答案仅供参考，具体的推荐产品和产品介绍链接地址可能需要根据实际情况进行调整。
相关·内容
我的问题与this one几乎相同，但显著的不同是在PyTorch中。我不喜欢使用Numpy解决方案，因为这将涉及到将数据移回CPU。我看到，和Numpy一样，PyTorch有一个nonzero函数，但是它的where函数(我链接的Numpy线程中的解决方案)的行为与Numpy的不同。我想要的行为是一个is_zero()函数，如下所示： >>>
浏览 586提问于2020-03-24得票数 0
回答已采纳
我在火把中有两个张量：和我如何使用pytorch的张量API(Ops)来查找tensorB中缺少的TensorA元素，然后对于每个缺失的元素，返回TensorA中最近的前一个元素的索引</
浏览 0提问于2023-04-13得票数 0
有没有什么方法可以检索二维pytorch张量中包含的最大元素的行和列索引？例如，请参阅下面的pytorch张量a：>> torch.tensor([1,2,3], [6,7,8])
张量a中最大的元素是9，它发生在第二行的第一列。如果我将其更改为从零开始的<
浏览 3提问于2020-02-27得票数 0
1回答
PyTorch是否支持类似dict的对象，通过这些对象我们可以反向传播梯度，比如PyTorch中的张量？
我的目标是计算一个大型矩阵中的几个元素(1%)的梯度。但是如果我使用PyTorch的标准张量来存储矩阵，我需要将整个矩阵保存在我的GPU中，这会导致由于在训练期间可用的G
浏览 6提问于2022-03-08得票数 2
我可以用相同形状的布尔数组/张量或者包含我想要的元素的整数索引的数组/张量来索引我的numpy数组/ pytorch张量。哪个更快？
浏览 12提问于2019-09-04得票数 3
我正在用PyTorch编写一些代码，我遇到了gather函数。查看我看到的文档，索引参数接受LongTensor，这是为什么？
浏览 81提问于2019-05-28得票数 0
回答已采纳
1回答
我有两个一维张量。一个是预测向量，第二个是标签向量。我正在尝试编写一个循环来检查向量之间的元素差异。如果发现了这样的差异，我想做另一个操作，为了简单起见，假设我想打印("Diff spotted")。到目前为止，我想出了这个，但我得到了一个错误:期望的对象是标量类型Byte，但得到了参数#2 'other‘的标量类型Float。我很感激你能帮上忙。也许有一些更有效的方法来做到这一点，没有循环。
浏览 1提问于2020-04-29得票数 0
我正在使用cudas运行一些代码，我需要在CPU上测试相同的代码以比较运行时间。为了决定是使用常规pytorch张量还是使用cuda浮点张量，我使用的库称为torch.cuda.is _ 可用()。有没有简单的方法让这个函数返回false？我查看了pytorch源代码，在我的例子中是torch.cuda.is _ 可获得的回报 torch._C._cuda_getDeviceCount() >
浏览 350提问于2018-10-24得票数 4
回答已采纳
1回答
tensor([[ 1, 2, 3, 4, 5], [11, 12, 13, 14, 15]])所以我做了c = a[:,[2]]，它给了我预期的答案，但在自动评分器上还是失败了。same_elements and same_storage
print('c correct:', check(a, c, [[3], [8],
浏览 2提问于2020-06-27得票数 1
回答已采纳
我使用的是PyTorch (1.8)。是否有一种聪明的方法可以在所有具有相同输出索引的数据点上获得一个元素级的最大值？假设我有一个大小的数据张量(N，M)，以及包含指数[0，K)的大小指数张量(N，)。现在我想根据索引值将数据张量放入一个大小的张量(K，M)，但是如果两个或多个数据点被绑定到相同的时隙中
浏览 8提问于2021-07-28得票数 2
回答已采纳
我是PyTorch的中级学习者，在最近的一些案例中，我看到人们使用torch.inference_mode()而不是著名的torch.no_grad()来验证经过训练的agent在强化学习(RL)实验中的有效性我检查了，它们有一个由两个标志组成的表，用于禁用梯度计算。老实说，如果我读到描述的话，对我来说听起来完全一样。有人找到解释了吗？
浏览 53提问于2022-10-25得票数 -1
1回答
对火炬张量的迭代
、、、
什么是最好和最快的方式来迭代张量。令人费解的是，为什么我要得到张量而不是值。x in t]但却期望这样的行为t.numpy() ]如果可能的话，我宁愿不改做胖的？
浏览 2提问于2022-09-15得票数 2
回答已采纳
1回答
我有以下形状的PyTorch张量：(100，5,100)。我需要将它转换成形状张量( 100,100)，从每一行中只选择第二维中的一个项，这意味着在这5个元素中，我只需要一个，与其对应的100个元素。为了执行这个操作，我有第二个形状张量(100 )，它的索引指定在每一行中应该选择哪一项。
有没有一种简单的方法来
浏览 5提问于2022-04-21得票数 1
回答已采纳
Pytorch是否有与Numpy的元素级 (logical_and、logical_or、logical_not和logical_xor)等效的and？在使用CPU时，对Pytorch张量调用Numpy函数似乎工作得足够好，甚至可以生成Pytorch张量作为输出。我问主要是因为我假设如果pytorch计算在GPU中运行，这将不会很好地工作。我在中查看了所有包含字符
浏览 1提问于2019-02-08得票数 5
回答已采纳
我有一个2D张量，每一行都有一些非零元素，如下所示： import torch
tmp = torch.tensor([[0, 0, 1, 0, 1, 0, 0],[0, 0, 0, 1, 1, 0, 0]], dtype=torch.float) 我想要一个包含每行第一个非零元素索引的张量： indices = tensor([2],[3]) 我如何在Pytorch中计算它？
浏览 175提问于2019-05-11得票数 7
回答已采纳
3回答
假设我有张量 t = torch.tensor([1,2,3,4,5]) 我想使用一个相同大小的索引张量来拆分它，它告诉我每个元素应该放在哪个拆分中。indices = torch.tensor([0,1,1,0,2]) 所以最终的结果是 splits
[tensor([1,4]), tensor([2,3]), tensor([5])] 在Pytorch中有没有一种很好的方法来做到这一点？编辑:通常会有2到3个以上
浏览 57提问于2021-03-22得票数 2
1回答
我第一次尝试PyTorch，遇到了一些问题。我在下面分享了我的一些代码，并有两个问题。
Q1:我的输出大小应该是多少？每个输入应该导致一个输出，等于6个可能的输出标签(1-6)中的一个。Q2:我的精确度计算有问题，我认为这与Q1有关。predicted最终的大小是4 x 4411，其中4是我的批处理大小(所以我认为这是正确的)，但是4411是我的特征/输入大小，这几乎可以肯定是错误
浏览 0提问于2021-04-08得票数 1
我是PyTorch新手，正在尝试重现这个项目：https://github.com/eXascaleInfolab/ActiveLink 然而，feedforward()中出现了错误，这几天一直困扰着我，这里是代码的一部分(有关该模型的完整代码，请参阅https://github.com/eXascaleInfolab/ActiveLink/blob/master/models.py )： def forward我使用调试器试图找出
浏览 305提问于2020-08-21得票数 0
回答已采纳
1回答
因此，有一个张量为(N，M，C)。让我们称其为特征。 我通过M维计算最大元素和索引，以找到每个C维的最大点(最大池化操作)，结果最大张量(N，1，C)和索引张量(N，1，C)。18, 19] [[1, 2, 3, 2, 1, 0]]] 例如，索引中的第一个元素<
浏览 22提问于2021-03-27得票数 1
回答已采纳
1回答
我有一个PyTorch张量a，形状如下：a = torch.tensor([[[1., 0., 0., 0.]],
[[0., 1., 0., 0.]],张量a的每一行都有4个元素，1和0，假设I相应地索引这个张量的行和列。例如，行0(最上面的行)中的条目是[[1., 0., 0., 0.]]，而第3列(最右边的列)中的
浏览 1提问于2020-08-09得票数 1
回答已采纳
扫码
添加站长 进交流群
领取专属 10元无门槛券
手把手带您无忧上云
相关资讯
热门标签
云服务器
ICP备案
实时音视频
即时通信 IM
对象存储活动推荐
腾讯云开发者
