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