Row major indexing

I just want to be sure: tensor.index({0,3}) is the top element of the fourth column of the tensor?


Yes, the indexing would use the row, then column to grab the value.
As a verification you could print the complete tensor as well as the indexed value and compare them.

1 Like