I just want to be sure: tensor.index({0,3}) is the top element of the fourth column of the tensor?
Thanks.
I just want to be sure: tensor.index({0,3}) is the top element of the fourth column of the tensor?
Thanks.
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.