View page source

warp.get_diag#

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

  • Kernel

  • Differentiable

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