如何在TensorFlow中獲得None的真實形狀（動態輸入形狀）？

Question

我有一個占位符，其形狀為[無，尺寸]，“無”表示批處理大小。 我想得到“無”的真實形狀。

構建模型時，我嘗試了兩種方法：

首先，x.get_shape（）並以[Dimension（None），Dimension（128）]的形式獲取形狀

其次，x.shape並獲得形狀為[Dimension（None），Dimension（128）]

我想要的是真實形狀，例如，當本輪的批次大小為100時，我想獲得[Dimension（100），Dimension（128）]。

如何獲得動態輸入形狀？

Answer 1

我相信tf.shape是您想要的。

在會話運行時， 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來初始化新變量。