torch.take¶ torch.take(input, index) → Tensor¶ 返回一个新的张量,其中包含 input 中给定索引处的元素。输入张量被视为一维张量。结果采用与索引相同的形状。 参数 input (Tensor) – 输入张量。 index (LongTensor) – 张量中的索引 示例 >>> src = torch.tensor([[4, 3, 5], ... [6, 7, 8]]) >>> torch.take(src, torch.tensor([0, 2, 5])) tensor([ 4, 5, 8])