Класс Zeros
Класс Zeros создает инициализатор,
который заполняет тензор значениями 0.
Первый параметр задает тип данных элементов.
Инициализатор применяется к форме тензора
через метод __call__, которому первым
параметром передается форма shape,
а вторым - тип данных.
Синтаксис
tf.keras.initializers.Zeros()
Пример
Давайте создадим инициализатор нулями и применим его к форме из пяти элементов:
import tensorflow as tf
init = tf.keras.initializers.Zeros()
t = init(shape=(5,))
print(t)
Результат выполнения кода:
tf.Tensor([0. 0. 0. 0. 0.], shape=(5,), dtype=float32)
Пример
Давайте создадим инициализатор нулями с
типом int32 и применим его к форме
из трех строк и двух столбцов:
import tensorflow as tf
init = tf.keras.initializers.Zeros(dtype=tf.int32)
t = init(shape=(3, 2))
print(t)
Результат выполнения кода:
tf.Tensor(
[[0 0]
[0 0]
[0 0]], shape=(3, 2), dtype=int32)
Пример
Давайте используем инициализатор нулями в
слое Dense для весов:
import tensorflow as tf
layer = tf.keras.layers.Dense(
units=3,
kernel_initializer=tf.keras.initializers.Zeros()
)
t = tf.constant([[1, 2, 3], [4, 5, 6]])
res = layer(t)
print(res)
Результат выполнения кода:
tf.Tensor(
[[0. 0. 0.]
[0. 0. 0.]], shape=(2, 3), dtype=float32)
Смотрите также
-
класс
Ones,
который создает инициализатор единицами -
класс
Constant,
который создает инициализатор константой -
класс
RandomNormal,
который создает инициализатор нормальным распределением -
класс
GlorotUniform,
который создает инициализатор Глоро