cupy.diagonal#

cupy.diagonal(a, offset=0, axis1=0, axis2=1)[source]#

Returns specified diagonals.

This function extracts the diagonals along two specified axes. The other axes are not changed. This function returns a writable view of this array as NumPy 1.10 will do.

Parameters
  • a (cupy.ndarray) – Array from which the diagonals are taken.

  • offset (int) – Index of the diagonals. Zero indicates the main diagonals, a positive value upper diagonals, and a negative value lower diagonals.

  • axis1 (int) – The first axis to take diagonals from.

  • axis2 (int) – The second axis to take diagonals from.

Returns

A view of the diagonals of a.

Return type

cupy.ndarray

See also

numpy.diagonal()