PyTorch Forums
Does torch.index_select support the input index is 2-dimension?
linyu
(linyu)
August 15, 2018, 2:04am
1
Does torch.index_select support the input index is 2-dimension?