tf.keras.initializers.Initializer

Базовый класс инициализатора: все инициализаторы Keras наследуются от этого класса.

Инициализаторы должны реализовывать метод __call__ со следующим сигнатурой:

def __call__(self, shape, dtype=None, **kwargs):
  # returns a tensor of shape `shape` and dtype `dtype`
  # containing values drawn from a distribution of your choice.

По желанию, вы также можете реализовать метод get_config и метод класса from_config, чтобы поддерживать сериализацию — точно так же, как и с любым объектом Keras.

Вот простой пример: инициализатор случайного нормального распределения.

import tensorflow as tf

class ExampleRandomNormal(tf.keras.initializers.Initializer):

  def __init__(self, mean, stddev):
    self.mean = mean
    self.stddev = stddev

  def __call__(self, shape, dtype=None, **kwargs):
    return tf.random.normal(
        shape, mean=self.mean, stddev=self.stddev, dtype=dtype)

  def get_config(self):  # To support serialization
    return {"mean": self.mean, "stddev": self.stddev}

Обратите внимание, что нам не нужно реализовывать from_config в примере выше, так как аргументы конструктора класса — ключи в конфигурации, возвращаемой get_config, совпадают. В этом случае, по умолчанию from_config работает нормально.

Методы

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
)

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