aesara.tensor.diagonal

aesara.tensor.diagonal(a, offset=0, axis1=0, axis2=1)

A helper function for ExtractDiag. It accepts tensor with ndim >= 2 as input. The name diagonal is just meant to keep it consistent with numpy.

Parameters:

• a (symbolic tensor) –

• offset (int) – offset

• axis1 (int) –

• axis2 (int) –

Returns:

tensor

Return type:

symbolic tensor