Problem about the function torch.index_select(input, dim, index, out(05) Save Image Image gallery: Torch Index Select