python - 如何在 TensorFlow 中获取 None 的真实形状(动态输入形状)？

Question

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

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

首先，x.get_shape() 并获取形状为 [Dimension(None), Dimension(128)]

其次，x.shape 并得到 shape 为 [Dimension(None), Dimension(128)]

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

如何获取动态输入形状？

Answer 1

我相信tf.shape正是您要找的。

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

完整示例如下:

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

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