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

>>阅读原文<<