代码之家 › 专栏 › 技术社区 › Thermodynamix

PyTorch中某些索引的复制张量元素

indices torch tensor pytorch machine-learning

Thermodynamix · 技术社区 · 3 年前

所需的操作在精神上类似于 torch.Tensor.index_copy ,但有点不同。

最好用一个例子来解释。

张量 A 具有我们将复制的原始值:

[10, 20, 30]

张量 B 具有的索引 A. :

[0, 1, 0, 1, 2, 1]

张量 C 长度与 B ,包含的索引值 A. :

[10, 20, 10, 20, 30, 20]

什么是制作的好方法 C 从…起 A. 和 B 在PyTorch中,不使用循环?

1 回复 | 直到 3 年前

jhso 3 年前

你试过只按A索引吗?

In [1]: import torch
  
In [2]: a = torch.tensor([20,30,40])

In [3]: b = torch.tensor([0,1,2,1,1,2,0,0,1,2])

In [4]: a[b]
Out[4]: tensor([20, 30, 40, 30, 30, 40, 20, 20, 30, 40])

推荐文章

Qinqing Liu · 如何构造一个三维张量,其中每个二维子张量是PyTorch中的对角矩阵?

7 年前

Elias E. · pytorch神经网络中的交叉熵损失。交叉熵()

7 年前

Wasi Ahmad · 参数维数对聚集函数的影响

7 年前

Salih Karagoz · 如何为图像命名。显示窗口?

7 年前

Beanocean · hdf5缺少依赖项:图腾错误:克隆git存储库失败;git clome超时

8 年前

floyd · 为什么我不能使用转置函数没有:?

8 年前

ytrewq · 从火炬中删除项目。张量

8 年前

Minkyu Choi · 当有两个gpu时,如何设置火炬仅使用一个gpu?

9 年前

Amir · 如何计算火炬中相对于任意层/重量的损失梯度?

9 年前

Alejandro Simkievich · 如何从python获取lua/翻译中的类类型

9 年前