LongTensor([0, 2]) r = torch.index_select(v, 1, indices) # Select element 0 and 2 for each dimension 1. # Masked select # 0 0 0 # 1 1 1 # 1 ...
確定! 回上一頁