Нулевой инициализатор для смещений, использующих 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())