https://am-press.github.io/posts/maths/tensor-trace/