Long Tensor([0,0]) print torch.index_select (t, 0, index) # 1 2 # 1 2 print ... 3 # 4 # 5 # 6 # Get Indices of Non-Zero elements print torch. nonzero(torch.
確定! 回上一頁