PyTorch tensor 翻转
June 18, 2019, 5:33 p.m.
read: 2881
转自https://discuss.pytorch.org/t/how-to-reverse-a-torch-tensor/382 形状为(10)的tensor翻转 tensor = torch.rand(10) # your tensor # create inverted indices idx = [i for i in range(tensor.size(0)-1, -1, -1)] idx = torch.LongTensor(idx) inverted_tensor = tensor.index_select(0, idx) 形状为(1,10,1)的tensor翻转 tensor…