warp.get_diag#

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

  • Kernel

  • Differentiable

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