tf.compat.v1.truncated_normal_initializer
Инициализатор, генерирующий усечённое нормальное распределение.
tf.compat.v1.truncated_normal_initializer(
mean=0.0,
stddev=1.0,
seed=None,
dtype=tf.dtypes.float32
)
Переход на TF2
Хотя это и устаревший compat.v1 API, этот символ совместим с выполнением в режиме eager и tf.function.
Чтобы перейти на TF2, переключитесь на использование либо tf.initializers.truncated_normal или tf.keras.initializers.TruncatedNormal (ни один из которых не из compat.v1) и передайте dtype при вызове инициализатора. Имейте в виду, что значения по умолчанию для stddev и поведение фиксированных семян изменились.
Структурное отображение на TF2
До:
initializer = tf.compat.v1.truncated_normal_initializer( mean=mean, stddev=stddev, seed=seed, dtype=dtype) weight_one = tf.Variable(initializer(shape_one)) weight_two = tf.Variable(initializer(shape_two))
После:
initializer = tf.initializers.truncated_normal( mean=mean, seed=seed, stddev=stddev) weight_one = tf.Variable(initializer(shape_one, dtype=dtype)) weight_two = tf.Variable(initializer(shape_two, dtype=dtype))
Как сопоставить аргументы
| Имя аргумента TF1 | Имя аргумента TF2 | Примечание |
|---|---|---|
mean | mean | Без изменений по умолчанию |
stddev | stddev | Значение по умолчанию изменилось с 1.0 на 0.05 |
seed | seed | |
dtype |
dtype | Родной API TF2 принимает его только как аргумент __call__, а не как аргумент конструктора. |
partition_info | - | (__call__ аргумент в TF1) Не поддерживается |
Описание
Эти значения похожи на значения из random_normal_initializer, за исключением того, что значения, которые на более чем на две стандартные отклонения от среднего, отбрасываются и перегенерируются. Это рекомендуемый инициализатор для весов и фильтров нейронных сетей.
| Аргументы | |
|---|---|
mean | скаляр Python или скалярный тензор. Среднее значение случайных значений, которые нужно сгенерировать. |
stddev | скаляр Python или скалярный тензор. Стандартное отклонение случайных значений, которые нужно сгенерировать. |
seed | Целое число Python. Используется для создания случайных семян. См. tf.compat.v1.set_random_seed для получения информации о поведении. |
dtype | Тип данных по умолчанию, используемый, если аргумент dtype не указан при вызове инициализатора. Поддерживаются только типы с плавающей точкой. |
Методы
from_config
@classmethod
from_config(
config
)
Инициализирует инициализатор по словарю конфигурации.
Пример:
initializer = RandomUniform(-1, 1) config = initializer.get_config() initializer = RandomUniform.from_config(config)
| Аргументы | |
|---|---|
config | Словарь Python. Обычно это результат get_config. |
| Возвращаемое значение | |
|---|---|
| Экземпляр инициализатора. |
get_config
get_config()
Возвращает конфигурацию инициализатора в виде сериализуемого в JSON словаря.
| Возвращаемое значение | |
|---|---|
| Сериализуемый в JSON словарь Python. |
__call__
__call__(
shape, dtype=None, partition_info=None
)
Возвращает тензор, инициализированный в соответствии с заданным инициализатором.
| Аргументы | |
|---|---|
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/api_docs/python/tf/compat/v1/truncated_normal_initializer