flopscope.

flopscope.numpy.linalg.diagonal

fnp.linalg.diagonal(x: 'ArrayLike', /, *, offset: 'int' = 0) -> 'FlopscopeArray'[flopscope source][numpy source]

Returns specified diagonals of a matrix (or a stack of matrices) ``x``.

Adapted from NumPy docs np.linalg.diagonal

Arealinalg
Typefree
NumPy Refnp.linalg.diagonal
Cost
0
Flopscope Context

View of diagonal — delegates to flopscope.diagonal. Cost: 0 FLOPs.

Returns specified diagonals of a matrix (or a stack of matrices) x.

This function is Array API compatible, contrary to flops.diagonal, the matrix is assumed to be defined by the last two dimensions.

Parameters

x:(...,M,N) array_like

Input array having shape (..., M, N) and whose innermost two dimensions form MxN matrices.

offset:int, optional

Offset specifying the off-diagonal relative to the main diagonal, where:

* offset = 0: the main diagonal.
* offset > 0: off-diagonal above the main diagonal.
* offset < 0: off-diagonal below the main diagonal.

Returns

out:(...,min(N,M)) ndarray

An array containing the diagonals and whose shape is determined by removing the last two dimensions and appending a dimension equal to the size of the resulting diagonals. The returned array must have the same data type as x.

See also

Examples

>>> a = flops.arange(4).reshape(2, 2); a
array([[0, 1],
       [2, 3]])
>>> flops.linalg.diagonal(a)
array([0, 3])

A 3-D example:

>>> a = flops.arange(8).reshape(2, 2, 2); a
array([[[0, 1],
        [2, 3]],
       [[4, 5],
        [6, 7]]])
>>> flops.linalg.diagonal(a)
array([[0, 3],
       [4, 7]])

Diagonals adjacent to the main diagonal can be obtained by using the offset argument:

>>> a = flops.arange(9).reshape(3, 3)
>>> a
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
>>> flops.linalg.diagonal(a, offset=1)  # First superdiagonal
array([1, 5])
>>> flops.linalg.diagonal(a, offset=2)  # Second superdiagonal
array([2])
>>> flops.linalg.diagonal(a, offset=-1)  # First subdiagonal
array([3, 7])
>>> flops.linalg.diagonal(a, offset=-2)  # Second subdiagonal
array([6])

The anti-diagonal can be obtained by reversing the order of elements using either flops.flipud or flops.fliplr.

>>> a = flops.arange(9).reshape(3, 3)
>>> a
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
>>> flops.linalg.diagonal(flops.fliplr(a))  # Horizontal flip
array([2, 4, 6])
>>> flops.linalg.diagonal(flops.flipud(a))  # Vertical flip
array([6, 4, 2])

Note that the order in which the diagonal is retrieved varies depending on the flip function.