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.