espnet2.enh.layers.ncsnpp_utils.layers.variance_scaling

source

espnet2.enh.layers.ncsnpp_utils.layers.variance_scaling(scale, mode, distribution, in_axis=1, out_axis=0, dtype=torch.float32, device='cpu')

Ported from JAX.