warp.trace# warp.trace(a: Matrix[Scalar, Any, Any]) → Scalar# Kernel Differentiable Compute the trace of matrix a.