DeltaOrthogonal

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

Construct an initializer for delta orthogonal kernels; see arXiv:1806.05393.

The shape must be 3D, 4D or 5D.