pytorch-(torch.take())根据索引返回指定索引上的数据集合
torch.take(input, index)->Tensor返回一个新的张量,其中的元素是输入元素在给定的索引处,将输入张量视为视为一维张量。结果tensor的形状与索引相同。参数介绍:input:输入tensor。indices:索引接下来看用法:src = torch.tensor([[4, 3, 5], [6, 7, 8]])torch.take(src, torch.tensor([0, 2, 5]))tensor([ 4, 5,