Spec-Zone.ru › TensorFlow 1.15

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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API