espnet2.enh.layers.bsrnn.choose_norm

source

espnet2.enh.layers.bsrnn.choose_norm(norm_type, channel_size, shape='BDTF')

The input of normalization will be (M, C, K), where M is batch size.

C is channel size and K is sequence length.