TensorFlow – 变量

变量存储可变化的值,例如可用于存储可训练参数:权重和偏置。

要创建一个变量,可以使用tf.get_variable()方法。

tf.get_variable(name = "", values, dtype, initializer)

参数

  • name = "": 变量名称
  • values: 指定变量形状
  • dtype: 数据类型,可选
  • initializer: 如何初始化张量,可选。如果指定了初始化器,则不需要包含“values”,因为使用的是“初始化器”的形状。

例如,下面的代码创建了一个形状是1×2的变量,变量中元素的默认值是随机值,变量命名为var。

import tensorflow as tf
# 创建一个变量
var = tf.get_variable("var", [1, 2])
print(var.shape)

输出

(1, 2)      

把变量中的元素值初始化为0

import tensorflow as tf
var_init_1 = tf.get_variable("var_init_1", [1, 2], dtype=tf.int32,  initializer=tf.zeros_initializer)
print(var_init_1.shape)     

输出

(1, 2)      

可以使用张量常量初始化变量。

# 创建一个2x2矩阵常量
tensor_const = tf.constant([[10, 20],[30, 40]])
# 使用矩阵常量初始化变量
var_init_2 = tf.get_variable("var_init_2", dtype=tf.int32,  initializer=tensor_const)
print(var_init_2.shape)         

输出

(2, 2)      


浙ICP备17015664号 浙公网安备 33011002012336号 联系我们 网站地图  
@2019 qikegu.com 版权所有,禁止转载