对于给定的2D张量，我想检索值为1的所有索引。我希望能够简单地使用torch.nonzero(a == 1).squeeze()，这将返回tensor([1, 3, 2])。然而，相反，torch.nonzero(a == 1)返回一个2D张量(没关系)，每行有两个值(这不是我所期望的)。然后，返回的索引应该用于对三维张量的第二维(索引1)进行索引</em
浏览 7提问于2019-09-27得票数 1
回答已采纳
我有一个形状为(x, n)的火炬张量a和另一个形状为y <= x的张量b。b的每一列都包含a的一系列行索引，我想要做的是用b以某种方式对a进行索引，这样我就可以获得一个形状为(y, n)的张量，其中第i列包含a[:, i][b[:, i]] (不太确定这是否是正确的表达方式)我正在寻找的是一种方法，它不会产生张量c，但也让
浏览 15提问于2019-04-26得票数 0
我有一个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
回答已采纳
我有一个2D张量和一个索引张量。2D张量有一个批次维度和一个具有3个值的维度。我有一个索引张量，它只选择3个值中的1个元素。产生只包含索引张量中元素的切片的“最佳”方法是什么？
浏览 20提问于2019-02-21得票数 0
回答已采纳
假设我有一个具有形状(大小，大小)的2D张量，我想得到两个新的张量，其中包含原始张量的行索引和列索引。所以如果size是2，我想得到[0，0，1，1]和[0，1，0，1]
棘手的是，size是另一个张量，它的值只有在tensorflow会话中运行图形时才能知道。
浏览 33提问于2017-08-09得票数 2
回答已采纳
我想要插入一个一维火炬张量到一个特定的行号到一个2D火炬张量(使用Pytorch)。当我说我想使用Pytorch时，我不想把任何东西转换成一个
浏览 1提问于2021-08-01得票数 1
回答已采纳
有没有什么方法可以检索二维pytorch张量中包含的最大元素的行和列索引？例如，请参阅下面的pytorch张量a：>> torch.tensor([1,2,3], [6,7,8])
张量a中最大的元素是9，它发生在第二行的第一列。如果我将其更改为从零开始的pytho
浏览 3提问于2020-02-27得票数 0
1回答
有没有一种方法可以索引稀疏张量的单元/部分，就像在密集张量的情况下？# dens_tensor.shape = [10, 10, 10]如果没有，是否有人知道索引稀疏张量的变通方法？
浏览 0提问于2016-12-16得票数 2
2回答
我试图通过一个索引矩阵来访问一个火炬张量，最近我发现了这个代码，我找不到为什么它不起作用的原因。
下面的代码分为两部分。上半场证明是有效的，而第二次旅行则是错误的。我不明白为什么。
浏览 20提问于2022-02-18得票数 3
回答已采纳
我在PyTorch中发现了一行用于将3D张量简化为2D张量的代码。3D张量x的大小为torch.Size([500, 50, 1])，下面这行代码：用于将x简化为大小为torch.Size([50, 1])的2D张量。lengths也是包含值的shape torch.Size([50])的</
浏览 1提问于2020-02-13得票数 3
假设我们有形状(n, k)和(n, k)的两个2D张量。我们希望将这两个张量与所有行式排列连接在一起，以使得到的张量具有形状(n, n, 2*k)。例如， A = [[a, b], [c, d]]; B = [[e, f], [g, h]] 得到的张量应该是： [[[a, b, e, f], [a, b, g, h]], [[c, d, e, f], [c, d, g, h]]] 假设输入张量A和B
浏览 22提问于2020-08-25得票数 1
回答已采纳
1回答
假设我有一个浮点型的2D张量A和一个整型的1D张量B。B中的数字表示A的行的索引，如何在tensorflow图中有效地执行此索引的查找？
浏览 2提问于2017-11-09得票数 1
我试图根据另外两个张量的内容组装一个张量，如下所示：
我有一个叫A的二维张量，它的形状为I* J，另一个叫B的二维张量，它的形状为M* N，其元素是A的第一维的指数，我想得到一个具有M*N*J形状的三维张量C，使得[0，M)中的所有m和[0，N]中的所有m都有C[m,n,:] == A[B[m,n],:]。我可
浏览 3提问于2022-01-31得票数 1
回答已采纳
1回答
假设我有一个2D张量，如下所示：..以及我想要选择的行索引列表，它看起来类似于这个[0, 34, 100, ..., 745如何遍历并创建一个仅包含数组中包含其索引的行的新张量？
浏览 2提问于2020-02-03得票数 2
回答已采纳
这个问题类似于已经回答过的，但这个问题并不涉及如何检索多个元素的索引。
我有一个2D张量points，它有许多行和少数列，并且希望得到一个张量，其中包含该张量中所有元素的行索引。我事先知道points中存在哪些元素；它包含介于0到999之间的整数元素，我可以使用range函数来生成一个张量来反映可能的元素集。元素可以在任何列中。如何以避免循环或使用numpy<em
浏览 1提问于2020-02-05得票数 0
回答已采纳
我需要一个与tf.one_hot提供的1有些类似的张量，但我(有时)需要在每一行中有多个张量。更具体地说，给定一个2d张量A，当且仅当A包含列索引作为同一行中的值时，我需要一个在某个位置具有1的2d张量。
更糟糕的是，我给出的张量A是动态大小和填充的，但另一个1d张量提供了实
浏览 2提问于2017-05-13得票数 0
我有Pytorch 2d张量，它具有正态分布。是否有一种快速的方法使用Python来取消这个张量的10%的最大值？我认为这里有两种可能的方法：但这些看起来都不够快。那么，将张量的X最大值设置为零的最快方法是什么？
浏览 3提问于2021-12-17得票数 2
回答已采纳
1回答
这似乎是一件非常简单的事情，但我在任何地方都找不到一个很好的答案。假设我有一个2D火炬张量x [3, 4], [1, 2]]) [False, False], [ True, Tr
浏览 0提问于2022-07-19得票数 0
1回答
在使用pytorch时，我可以使用numpy数组吗？
我正在将一段代码从tensorflow转换为pytorch，代码在计算过程中使用numpy数组。在计算过程中，我可以将输入保持为numpy数组吗?还是必须将它们转换为torch张量？
浏览 0提问于2021-07-13得票数 0
扫码
添加站长 进交流群
领取专属 10元无门槛券
手把手带您无忧上云
相关资讯
热门标签
云服务器
ICP备案
对象存储
腾讯会议
云直播活动推荐
腾讯云开发者
