mindscience.common.SpectralNorm

class mindscience.common.SpectralNorm(module, n_power_iterations=1, dim=0, eps=1e-12)[源代码]

对给定模块中的参数应用谱归一化。谱归一化通过使用谱范数对权重张量进行重标定,从而稳定生成对抗网络(GAN)中判别器(critic)的训练过程。

参数:
  • module (nn.Cell) - 包含目标参数的模块。

  • n_power_iterations (int, 可选) - 用于计算谱范数的幂迭代次数。默认 1

  • dim (int, 可选) - 对应输出数量的维度。默认 0

  • eps (float, 可选) - 在计算范数时用于数值稳定性的 epsilon 值。默认 1e-12

输入:
  • input - 包含位置参数的输入。

  • kwargs - 其它关键字参数。

输出:

模块的前向推理结果。