[ 仅参数名不一致 ]torch.nn.utils.parametrizations.spectral_norm¶
torch.nn.utils.parametrizations.spectral_norm¶
torch.nn.utils.parametrizations.spectral_norm(module, name='weight', n_power_iterations=1, eps=1e-12, dim=None)
paddle.nn.utils.spectral_norm¶
paddle.nn.utils.spectral_norm(layer, name='weight', n_power_iterations=1, eps=1e-12, dim=None)
PyTorch 和 Paddle 功能一致，仅参数名不一致，具体如下：
参数映射¶
| PyTorch | PaddlePaddle | 备注 |
|---|---|---|
| module | layer | 要添加权重谱归一化的层，仅参数名不一致。 |
| name | name | 权重参数的名字。 |
| n_power_iterations | n_power_iterations | 将用于计算的 SpectralNorm 幂迭代次数。 |
| eps | eps | 用于保证计算中的数值稳定性。 |
| dim | dim | 重塑为矩阵之前应排列到第一个的维度索引。 |