1. 变量
- 变量可以在模型训练时加以更新,计算梯度等。一般地张量是不会计算梯度的(除非加入watch才能计算梯度)
- 创建变量用tf.Variable。tf2.x中取消了tf.get_variable
1.1 变量的创建
# 直接使用tf.Variable()创建
variable = tf.Variable([1.0, 2.0], trainable=True) # trainable设为False,可不进行梯度计算与更新
variable
>>>
微信扫一扫
# 直接使用tf.Variable()创建
variable = tf.Variable([1.0, 2.0], trainable=True) # trainable设为False,可不进行梯度计算与更新
variable
>>>
相关推荐