Skip to content

Add trace operation for tensors

Simon Praetorius requested to merge feature/operation-trace into main

Summary

This MR allows to compute the trace as the sum of the last two indices of a tensor. It is implemented in terms of the tensordot operation between the tensor and the identity-matrix. Not, the tensordot needs to be further improved to make this operation fast.

Merge request reports