warp.get_diag#

warp.get_diag(
mat: Matrix[Any, Any, Scalar],
) → Vector[Any, Scalar]#

  • Kernel

  • Differentiable

Returns a vector containing the diagonal elements of the square matrix mat.