tf.contrib.layers.variance_scaling_initializer
Возвращает инициализатор, который генерирует тензоры без масштабирования дисперсии.
tf.contrib.layers.variance_scaling_initializer(
factor=2.0, mode='FAN_IN', uniform=False, seed=None, dtype=tf.dtypes.float32
)
При инициализации глубокой сети желательно сохранять масштаб дисперсии входных данных постоянным, чтобы он не взрывался или не уменьшался к конечному слою. Этот инициализатор использует следующую формулу:
if mode='FAN_IN': # Count only number of input connections. n = fan_in elif mode='FAN_OUT': # Count only number of output connections. n = fan_out elif mode='FAN_AVG': # Average number of inputs and output connections. n = (fan_in + fan_out)/2.0 truncated_normal(shape, 0.0, stddev=sqrt(factor / n))
- Для получения Delving Deep into Rectifiers (также известного как "MSRA initialization"), используйте (по умолчанию):
factor=2.0 mode='FAN_IN' uniform=False - Для получения Convolutional Architecture for Fast Feature Embedding, используйте:
factor=1.0 mode='FAN_IN' uniform=True - Для получения Understanding the difficulty of training deep feedforward neural networks, используйте:
factor=1.0 mode='FAN_AVG' uniform=True. - Для получения
xavier_initializerиспользуйте либо:
factor=1.0 mode='FAN_AVG' uniform=True, или
factor=1.0 mode='FAN_AVG' uniform=False.
| Аргументы | |
|---|---|
factor | Вещественное число. Множительный фактор. |
mode | Строка. 'FAN_IN', 'FAN_OUT', 'FAN_AVG'. |
uniform | Использовать ли равномерное или нормальное случайное начальное значение. |
seed | Целое число Python. Используется для создания случайных семян. См. tf.compat.v1.set_random_seed для поведения. |
dtype | Тип данных. Поддерживаются только типы с плавающей точкой. |
| Возвращает | |
|---|---|
| Инициализатор, который генерирует тензоры с единичной дисперсией. |
| Возможные исключения | |
|---|---|
ValueError | если dtype не является типом с плавающей точкой. |
TypeError | если mode не входит в ['FAN_IN', 'FAN_OUT', 'FAN_AVG']. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/layers/variance_scaling_initializer