定义“学习参数”的变量

Posted 2021-02-11 wqh2216

学习参数的定义与输入的定义很像，分为直接定义和字典定义两种方法，如果参数过多的话，那么便会采用第二种定义。

直接定义：

1 W = tf.Variable(tf.random_normal([1]), name = "weight")
2 b = tf.Variable(tf.zeros([1]), name = "bias")

通过字典类型定义“学习参数”：

1 # 模型参数
2 paradict = {
3         ‘w‘: tf.Variable(tf.random_normal([1]),
4         ‘b‘: tf.Variable(tf.zeros([1]))
5 }
6 # 前向结构
7 z = tf.multiply(X, paradict[‘w‘]) + paradict[‘b‘]