diagonal(a, offset=0, axis1=0, axis2=1)¶
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