【问题标题】:How to get real shape of None (dynamic input shape) in TensorFlow?如何在TensorFlow中获得None(动态输入形状)的真实形状?
【发布时间】:2017-12-05 02:48:46
【问题描述】:

我有一个占位符,其形状为 [None, dimension],“None”表示批量大小。我想得到“无”的真实形状。

我在构建模型时尝试了两种方法:

首先,x.get_shape() 得到形状为 [Dimension(None), Dimension(128)]

第二,x.shape,得到shape为[Dimension(None), Dimension(128)]

而我想要的是真实的形状,比如本轮batch size为100时,我想得到[Dimension(100), Dimension(128)]。

如何获得动态输入形状?

【问题讨论】:

    标签: python tensorflow


    【解决方案1】:

    我相信tf.shape 就是您要找的。​​p>

    tf.shape(x) 可以在会话运行时获取形状。

    完整的例子如下:

    import tensorflow as tf
    a = tf.ones([3,4])
    b = tf.shape(a)
    sess=tf.Session()
    print(b.eval(session=sess))
    

    您也可以使用b 来初始化新变量。

    【讨论】:

    • tf.shape(x) return (2,),我觉得这个意思是维数。
    猜你喜欢
    • 2016-09-02
    • 1970-01-01
    • 2018-10-23
    • 2016-07-30
    • 2021-10-21
    • 1970-01-01
    • 2018-08-23
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多