[ 仅参数名不一致 ]torch.nn.utils.parametrizations.weight_norm

torch.nn.utils.parametrizations.weight_norm

torch.nn.utils.parametrizations.weight_norm(module, name='weight', dim=0)

paddle.nn.utils.weight_norm

paddle.nn.utils.weight_norm(layer, name='weight', dim=0)

其中,PyTorch 和 Paddle 仅参数名不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
module layer 要添加权重归一化的层,仅参数名不一致。
name name 权重参数的名字。
dim dim 进行归一化操作的切片所在维度。