在Tensorflow中,可以使用tf.trainable_variables()函数来打印可训练的所有变量。该函数会返回一个包含所有可训练变量的列表。
以下是一个示例代码:
import tensorflow as tf
# 创建一些变量
var1 = tf.Variable(tf.constant(1.0), name='var1')
var2 = tf.Variable(tf.constant(2.0), name='var2')
var3 = tf.Variable(tf.constant(3.0), name='var3')
# 打印可训练的所有变量
trainable_vars = tf.trainable_variables()
print("可训练的变量列表:")
for var in trainable_vars:
print(var.name)
# 输出结果:
# 可训练的变量列表:
# var1:0
# var2:0
# var3:0
在上述代码中,我们首先创建了三个变量var1、var2和var3。然后使用tf.trainable_variables()函数获取可训练的所有变量,并通过遍历打印出每个变量的名称。
注意:tf.trainable_variables()函数只会返回可训练的变量,即在定义变量时没有设置trainable=False的变量。
领取专属 10元无门槛券
手把手带您无忧上云