torch.Tensor.index_select torch.index_select — PyTorch 1.12 documentation torch.index_select torch. index_select ( input , dim , index , * , out = None ) → Tensor Returns a new tensor which indexes the input tensor along dimension dim using the entries in index which is a LongTensor . The returned tensor has the same number of dimensions as the original tensor ( input ). ... pytorch.org 원래의 Tensor에서 원하는 index에 해당하는 요소(값)만 따로 뽑아서 새로운 Tensor를 만들어준다.
Parameter input : Tensor를 입력으로 받는다. dim : index할...
