tf.keras.layers.SpectralNormalization
Выполняет спектральную нормализацию весов целевого слоя.
Наследуется от: Wrapper, Layer, Operation
tf.keras.layers.SpectralNormalization(
layer, power_iterations=1, **kwargs
)
Этот обертка контролирует постоянную Липшица весов слоя, ограничивая их спектральную норму, что может стабилизировать обучение GAN.
| Аргументы | |
|---|---|
layer | Экземпляр keras.layers.Layer, у которого есть либо атрибут kernel (например, Conv2D, Dense...) или атрибут embeddings (слой Embedding). |
power_iterations | int, число итераций во время нормализации. |
**kwargs | Базовые ключевые аргументы обертки. |
Примеры:
Обернуть keras.layers.Conv2D:
>>> x = np.random.rand(1, 10, 10, 1) >>> conv2d = SpectralNormalization(keras.layers.Conv2D(2, 2)) >>> y = conv2d(x) >>> y.shape (1, 9, 9, 2)
Обернуть keras.layers.Dense:
>>> x = np.random.rand(1, 10, 10, 1) >>> dense = SpectralNormalization(keras.layers.Dense(10)) >>> y = dense(x) >>> y.shape (1, 10, 10, 10)
Ссылка:
| Атрибуты | |
|---|---|
input | Возвращает тензор(ы) ввода(ов) символической операции. Возвращает только тензор(ы), соответствующий первому вызову операции. |
output | Возвращает тензор(ы) вывода(ов) слоя. Возвращает только тензор(ы), соответствующий первому вызову операции. |
Методы
from_config
@classmethod
from_config(
config, custom_objects=None
)
Создает слой по его конфигурации.
Этот метод является обратным к get_config, способным создать тот же слой из словаря конфигурации. Он не обрабатывает соединение слоев (обрабатывается Network), ни весы (обрабатывается set_weights).
| Аргументы | |
|---|---|
config | Словарь Python, обычно результат get_config. |
| Возвращает | |
|---|---|
| Экземпляр слоя. |
normalized_weights
normalized_weights()
Генерирует спектрально нормализованные веса.
Этот метод возвращает обновленное значение для self.kernel со спектрально нормализованным значением, так что слой готов к call().
symbolic_call
symbolic_call(
*args, **kwargs
)
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/layers/SpectralNormalization