是否有便捷的方法来进行频谱正常化? (Sonnet 版本: 2.0.0)

亲爱的 Deepminder: 对于生成对抗网络,它引入了称为频谱归一化(参考: https://arxiv.org/abs/1802.05957)。但是根据我的了解,我发现将其集成到 Sonnet 有一定难度。这是因为频谱归一化会改变权重。但是我检查了 snt.Conv2D,似乎权重由于内部管理而不可操作。旧版 Sonnet 似乎有解决方案(https://GitHub.com/deepmind/sonnet/issues/99),但我无法在 Sonnet 2.0.0 版中找到"custom_getter"和其他解决方案。因此,我是否可以得到一些关于这种情况的建议?非常感谢。