warp.trace#

warp.trace(a: Matrix[Any, Any, Scalar]) Scalar#
  • Kernel

  • Differentiable

Compute the trace of matrix a.