gpt4 book ai didi

pytorch - 如何在 pytorch 中选择某个维度上的单个索引?

转载 作者:行者123 更新时间:2023-12-03 04:44:10 25 4
gpt4 key购买 nike

假设我有一个形状为 [8, 12, 2] 的张量序列。现在我想为每个第一维选择该张量,从而产生形状为 [8, 2] 的张量。维度 1 上的选择由存储在形状 [8] 的长张量 indices 中的索引指定。

我尝试了这个,但是它为 sequences 中的每个第一个维度选择 indices 中的每个索引,而不是仅选择一个。

sequences[:, indices]

如何在没有缓慢且丑陋的 for 循环的情况下进行此查询?

最佳答案

sequences[torch.arange(sequences.size(0)), indices]

关于pytorch - 如何在 pytorch 中选择某个维度上的单个索引?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53986301/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com