Нулевой инициализатор для смещений, использующих get_variable в тензорном потоке

Код, который я изменяю, использует tf.get_variable для весовых переменных и tf.Variable для инициализации смещения. После некоторых поисков кажется, что get_variable всегда следует отдавать предпочтение из-за его мобильности в отношении обмена. Поэтому я попытался изменить переменную смещения на get_variable но не могу заставить его работать.

Оригинал: tf.Variable(tf.zeros([128]), trainable=True, name="b1")

Моя попытка: tf.get_variable(name="b1", shape=[128], initializer=tf.zeros_initializer(shape=[128]))

Я получаю сообщение о том, что форма не должна быть указана для констант. Но удаление фигуры выдает ошибку без аргументов.

Я очень новичок в tf так что я, наверное, неправильно понимаю что-то фундаментальное здесь. Спасибо за помощь в продвижении:)

1 ответ

Решение

Следующее должно работать:tf.get_variable(name="b1", shape=[128], initializer=tf.zeros_initializer())

Другие вопросы по тегам