mlx.nn.init.orthogonal

orthogonal(gain: float = 1.0, dtype: Dtype = mlx.core.float32) → Callable[[array], array]

An initializer that returns an orthogonal matrix.

參數:

• gain (float, optional) -- Scaling factor for the orthogonal matrix. Default: 1.0.

• dtype (Dtype, optional) -- Data type of the array. Default: float32.

回傳值:

An initializer that returns an orthogonal matrix with the same shape as the input.

回傳值型別:

Callable[[array], array]