tf.keras.initializers.HeUniform

Инициализатор He uniform с масштабированием дисперсии.

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

tf.keras.initializers.HeUniform(
    seed=None
)

Также доступен через сокращенную функцию tf.keras.initializers.he_uniform.

Генерирует выборки из равномерного распределения в пределах [-limit, limit], где limit = sqrt(6 / fan_in) (fan_in — количество входных единиц в тензоре весов).

Примеры:

# Standalone usage:
initializer = tf.keras.initializers.HeUniform()
values = initializer(shape=(2, 2))

# Usage in a Keras layer:
initializer = tf.keras.initializers.HeUniform()
layer = tf.keras.layers.Dense(3, kernel_initializer=initializer)

Ссылки:

He и др., 2015 (pdf)

Методы

from_config

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

@classmethod
from_config(
    config
)

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

Пример:

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

get_config

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

get_config()

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

__call__

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

__call__(
    shape, dtype=None, **kwargs
)

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