synax.SpecialOrthogonalMatrix.init_params

synax.SpecialOrthogonalMatrix.init_params(key: Key) jax.Array[source]

Sample initial parameters.

Parameters:

key – PRNG key.

Returns:

Parameters.