tf.trainable_variables(),

参考：https://blog.csdn.net/cerisier/article/details/86523446