[ 仅参数名不一致 ]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 | 进行归一化操作的切片所在维度。 |
