introduction to pytorch [ ][ ][ ][ ][ ]]in [ ] shape out[ ]torch size([ ]in [ ] torch split( , , in [ ] out[ ](tensor([[ ],[ ]])tensor([[ ],[ ]])tensor([[ ],[ ]])tensor([[ ],[ ]])tensor([[ ],[ ]])the index_select method allows you to extract parts of tensor along given dimension note that the method takes three argumentsthe tensor to operate onthe dimension along which to extract dataand the tensor containing the indices in listing - we construct tensorand then extract data along each of the two dimensions listing - extracting parts of tensors using index_select in [ ] torch floattensor([[ , ],[ ][ ]]in [ ] out[ ]tensor([[ ][ ][ ]]