LU Kernel Pivoting Definition


Please let me know if this is not the right place to post this! I am having some trouble understanding the documentation for here:

Specifically, how is the returned pivots tensor defined? I thought that it would be the index of the pivot for each row during partial pivoting, but this is obviously not the case since the example given in the docs has [3,3,3] for the pivots. I could not find any additional documentation on this, so if someone has any ideas, please let me know!


1 Like