ClikaRT::ops::trace

function

trace()

Tensor trace(Tensor input)

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

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

Parameters

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

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

Throws

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

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