Revert "change tensor trace"

This reverts commit 8acb94f406564eafcb71d1e277f2f97698335700.