(1.12版本)如果要使用shape
必须要.shape.as_list()
,否则就会报错。
import tensorflow as tf
v = tf.get_variable('v',[1], initializer=tf.constant_initializer(1))
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
print(type(v.shape))
>><class 'tensorflow.python.framework.tensor_shape.TensorShape'>
网友评论