pytorch取值

    import torch a_data=torch.Tensor([1,2,3]) index=[0,1,1,2,1] ba=a_data[index] print(ba) 结果：python tensor([1., 2., 2., 3., 2.])数组 能够用数组取值，index中的每一个值都是a_data的索引，不能越界。code

>>阅读原文<<