I think this is implemented as the index_select function, you can try import torch A_idx = torch.LongTensor([0, 2]) # the index vector B ...
確定! 回上一頁