小编kon*_*nin的帖子

IndexError:使用列表进行索引时,维度 1 张量的索引过多

因此,我正在解决一个机器学习问题,在我的一个函数中,我尝试使用torch.tensor张量列表对 a 进行索引,有时索引有效,有时则无效。我知道此错误通常会发生的情况,即当尝试使用 2 个或更多索引对一维数组进行索引时,但当我使用索引列表进行索引时(根据我的理解),不应发生此错误。

因此,下面我将介绍发生在我身上的准确可重现的情况。我有两个由同一代码片段生成的张量列表。

>>> a
[tensor(7, device='cuda:0'), tensor(3, device='cuda:0'), tensor(5, device='cuda:0'), tensor(0, device='cuda:0'), tensor(4, device='cuda:0'), tensor(24, device='cuda:0'), tensor(27, device='cuda:0'), tensor(2, device='cuda:0'), tensor(6, device='cuda:0'), tensor(9, device='cuda:0'), tensor(18, device='cuda:0'), tensor(31, device='cuda:0'), tensor(1, device='cuda:0'), tensor(8, device='cuda:0')]
>>> b
[tensor(20319, device='cuda:0'), tensor(19303, device='cuda:0'), tensor(20181, device='cuda:0'), tensor(17466, device='cuda:0'), tensor(4941, device='cuda:0'), tensor(21132, device='cuda:0'), tensor(17381, device='cuda:0'), tensor(17332, device='cuda:0'), tensor(18432, device='cuda:0'), tensor(18205, device='cuda:0'), tensor(19689, device='cuda:0'), tensor(20743, device='cuda:0'), tensor(18466, device='cuda:0'), tensor(16804, device='cuda:0'), tensor(17521, device='cuda:0'), tensor(21147, device='cuda:0'), tensor(16152, device='cuda:0'), tensor(14863, device='cuda:0'), tensor(15458, device='cuda:0'), tensor(18847, device='cuda:0'), …
Run Code Online (Sandbox Code Playgroud)

python pytorch

5
推荐指数
1
解决办法
452
查看次数

标签 统计

python ×1

pytorch ×1