aesara.tensor.diagonal#

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

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