我有一个占位符，其形状为 [无，维度]，“无”表示批量大小。我想获得“无”的真实形状。
我在构建模型时尝试了两种方法：
首先，x.get_shape() 并获取形状为 [Dimension(None), Dimension(128)]
二、x.shape，获取shape为[Dimension(None), Dimension(128)]
而我想要的是真实的形状，比如这一轮batch size为100时，我想得到[Dimension(100), Dimension(128)]。
如何获得动态输入形状？
我相信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来初始化新变量。
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句