ClikaRT::ops::trace

function

trace()

Tensor trace(Tensor x)

Sum of the main diagonal of a rank-2 tensor.

$$\mathrm{tr}(x) = \sum_i x_{ii}$$

Parameters

• x: a rank-2 tensor [M, N].

Returns: a 0-D tensor with x's dtype.

Throws

• ClikaRT::Error: (INVALID_ARGUMENT) when x is not rank-2.

Declared in ClikaRT/compute/ops.h, line 4455