- 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.
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.
A view of the diagonals of
- Return type