tf.compat.v1.keras.initializers.he_uniform

Инициализатор, способный адаптировать свой масштаб к форме тензоров весов.

Наследуется от: VarianceScaling

tf.compat.v1.keras.initializers.he_uniform(
    seed=None
)

Переход к TF2

Хотя это устаревший compat.v1 API, этот символ совместим с выполнением eager и tf.function.

Чтобы перейти к API TF2, используйте либо tf.initializers.variance_scaling или tf.keras.initializers.VarianceScaling (ни один из них не из compat.v1) и передайте тип данных при вызове инициализатора.

Структурное соответствие TF2

До:

initializer = tf.compat.v1.variance_scaling_initializer(
  scale=scale,
  mode=mode,
  distribution=distribution
  seed=seed,
  dtype=dtype)

weight_one = tf.Variable(initializer(shape_one))
weight_two = tf.Variable(initializer(shape_two))

После:

initializer = tf.keras.initializers.VarianceScaling(
  scale=scale,
  mode=mode,
  distribution=distribution
  seed=seed)

weight_one = tf.Variable(initializer(shape_one, dtype=dtype))
weight_two = tf.Variable(initializer(shape_two, dtype=dtype))

Как сопоставить аргументы

Имя аргумента TF1	Имя аргумента TF2	Примечание
`scale`	`scale`	Без изменений по умолчанию
`mode`	`mode`	Без изменений по умолчанию
`distribution`	`distribution`	Без изменений по умолчанию. 'normal' отображается на 'truncated_normal'
`seed`	`seed`
`dtype`	`dtype`	API TF2 принимает его только как аргумент `__call__`, а не аргумент конструктора.
`partition_info`	-	(аргумент `__call__` в TF1) Не поддерживается

Описание

С distribution="truncated_normal" or "untruncated_normal", образцы извлекаются из усеченного/неусеченного нормального распределения со средним значением нуль и стандартным отклонением (после усечения, если оно используется) stddev = sqrt(scale / n), где n:

число входных единиц в тензоре весов, если mode = "fan_in"
число выходных единиц, если mode = "fan_out"
среднее значение числа входных и выходных единиц, если mode = "fan_avg"

С distribution="uniform", образцы извлекаются из равномерного распределения в пределах [-limit, limit] с limit = sqrt(3 * scale / n).

Args
`scale`	Коэффициент масштабирования (положительное число с плавающей точкой).
`mode`	Один из "fan_in", "fan_out", "fan_avg".
`distribution`	Используемое случайное распределение. Один из "normal", "uniform".
`seed`	Целое число Python. Используется для создания случайных семян. См. `tf.compat.v1.set_random_seed` для поведения.
`dtype`	Тип данных по умолчанию, используется, если аргумент `dtype` не предоставлен при вызове инициализатора. Поддерживаются только типы с плавающей точкой.

Возможные ошибки
`ValueError`	В случае неверного значения аргументов "scale", "mode" или "distribution".

Методы

`from_config`

Просмотреть исходный код

@classmethod
from_config(
    config
)

Инициализирует инициализатор из словаря конфигурации.

Пример:

initializer = RandomUniform(-1, 1)
config = initializer.get_config()
initializer = RandomUniform.from_config(config)

Args
`config`	Словарь Python. Обычно это результат `get_config`.

Возвращаемое значение
Экземпляр инициализатора.

`get_config`

Просмотреть исходный код

get_config()

Возвращает конфигурацию инициализатора в виде сериализуемого в JSON словаря.

Возвращаемое значение
Сериализуемый в JSON словарь Python.

`call`

Просмотреть исходный код

__call__(
    shape, dtype=None, partition_info=None
)

Возвращает тензор, инициализированный в соответствии с инициализатором.

Args
`shape`	Форма тензора.
`dtype`	Необязательный тип данных тензора. Если не указан, используется тип данных инициализатора.
`partition_info`	Необязательная информация о возможном разбиении тензора.

© 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/versions/r2.9/api_docs/python/tf/compat/v1/keras/initializers/he_uniform