warp.get_diag

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

• Kernel: true
• Python: true
• Differentiable: true

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