python What does the torch.gather and torch.index_select do? Stack(01) Save Image Image gallery: Torch.index_select