ClikaRT::ops::eye

function

eye()

Tensor eye(
    ScalarOrTensor n,
    std::optional<ScalarOrTensor> m = std::nullopt,
    DataType dtype = DataType::Float32,
    StreamOrDevice s = {}
)

A 2-D identity matrix [n, m]: 1 on the diagonal, 0 elsewhere; m defaults to n (square).

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