fntrace→Tensortrace(input: Tensor)sourceeditImplementing kernelC++ engine symbols that back this Python API.C++trace_opfree fnSum of the main diagonal of a 2-D tensor.ParametersinputTensorA 2-D tensor; it need not be square.ReturnsTensor0-d tensor holding ∑iinput[i,i]\sum_i \text{input}[i, i]∑iinput[i,i].Examples>>> import lucid >>> lucid.trace(lucid.tensor([[1.0, 2.0], [3.0, 4.0]])) tensor(5.)
