Solved by ptrblck in post #2 
              
              
                I’m unsure about the relative parity between numpy’s and PyTorch’s indexing, but I’m usually also referring to the numpy docs in necessary and wasn’t running into an unsupported use case so far.