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