Function marian::inits::eye

Function Documentation

Ptr<NodeInitializer> marian::inits::eye(float value = 1.f)

Set diagonal of two dimensional quadratic matrix to value.

Sets all values of the tensor to 0 and initializes the diagonal with the given value. If no value is specified 1 is used by default.

Return

A NodeInitializer