espnet2.enh.layers.ncsnpp_utils.up_or_down_sampling.get_weight

source

espnet2.enh.layers.ncsnpp_utils.up_or_down_sampling.get_weight(module, shape, weight_var='weight', kernel_init=None)

Get/create weight tensor for a convolution or fully-connected layer.