tf.Variable代码举例
时间: 2024-03-12 18:20:38 浏览: 102
代码示例
以下是一个使用tf.Variable定义变量的示例代码:
```
import tensorflow as tf
# 定义一个名为weights的变量,形状为[784, 10]
weights = tf.Variable(tf.zeros([784, 10]), name="weights")
# 初始化所有变量
init = tf.global_variables_initializer()
with tf.Session() as sess:
# 运行初始化操作
sess.run(init)
# 输出变量的值
print(sess.run(weights))
```
在这个例子中,我们使用`tf.Variable`定义了一个名为weights的变量,它的形状为[784, 10],即一个784行10列的矩阵。我们使用`tf.zeros`函数将其初始化为0。然后,我们使用`tf.global_variables_initializer`函数初始化所有变量,最后在会话中运行该操作,并输出变量的值。
阅读全文