The array API specification stipulates that the diagonal does not have to be the main one. This is controlled by the k parameter. Positive values correspond to upper diagonals whereas negative values correspond to lower diagonals. By default k=0 which is the main diagonal and thus our current behavior.
cc @gchanan @mruberry @rgommers @pmeier @asmeurer @leofang @AnirudhDagar @asi1024 @emcastillo @kmaehashi
The array API specification stipulates that the diagonal does not have to be the main one. This is controlled by the
kparameter. Positive values correspond to upper diagonals whereas negative values correspond to lower diagonals. By defaultk=0which is the main diagonal and thus our current behavior.cc @gchanan @mruberry @rgommers @pmeier @asmeurer @leofang @AnirudhDagar @asi1024 @emcastillo @kmaehashi