Класс Ones

Класс Ones создает инициализатор, который заполняет тензор единицами. Чаще всего используется для инициализации весов и смещений в слоях нейронной сети. Класс не принимает обязательных параметров, что делает его простым в использовании.

Синтаксис

tf.keras.initializers.Ones()

Пример

Давайте создадим инициализатор и применим его к тензору формы 2 на 3:

import tensorflow as tf

initializer = tf.keras.initializers.Ones()
t = initializer(shape=(2, 3))
print(t)

Результат выполнения кода:

tf.Tensor(
[[1. 1. 1.]
 [1. 1. 1.]], shape=(2, 3), dtype=float32)

Пример

Давайте используем инициализатор Ones в полносвязном слое Dense:

import tensorflow as tf

layer = tf.keras.layers.Dense(
	units=4,
	kernel_initializer=tf.keras.initializers.Ones(),
	bias_initializer=tf.keras.initializers.Ones()
)

t = tf.constant([[1.0, 2.0, 3.0]])
res = layer(t)
print(res)

Результат выполнения кода:

tf.Tensor([[7. 7. 7. 7.]], shape=(1, 4), dtype=float32)

Пример

Давайте создадим тензор формы 5 с помощью инициализатора Ones и выведем его форму:

<+python+> import tensorflow as tf initializer = tf.keras.initializers.Ones() t = initializer(shape=(5,)) print(t) print(t.shape) <-python+>

Результат выполнения кода:

tf.Tensor([1. 1. 1. 1. 1.], shape=(5,), dtype=float32)
(5,)

Смотрите также

класс Zeros,
который создает инициализатор, заполненный нулями
класс Constant,
который создает инициализатор с заданным значением
класс GlorotUniform,
который создает инициализатор с равномерным распределением
класс HeNormal,
который создает инициализатор с нормальным распределением