我正在修改的代码使用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
很新,所以我可能误解了一些基本的东西。感谢您提前的帮助:)
答案 0 :(得分:14)
以下应该工作:
tf.get_variable(name="b1", shape=[128], initializer=tf.zeros_initializer())