美文网首页
[tf]中的shape

[tf]中的shape

作者: VanJordan | 来源:发表于2018-12-24 09:15 被阅读11次

(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'>

相关文章

网友评论

      本文标题:[tf]中的shape

      本文链接:https://www.haomeiwen.com/subject/iqdakqtx.html