Orthogonal

class brainstate.nn.init.Orthogonal(scale=1.0, axis=-1, seed=None, unit=Unit('1'))

Construct an initializer for uniformly distributed orthogonal matrices.

If the shape is not square, the matrix will have orthonormal rows or columns depending on which side is smaller.