[ 仅参数名不一致 ]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 | 重塑为矩阵之前应排列到第一个的维度索引。 |