Question

使用validate_shape=False在tensorflow中创建变量时，它也会忽略变量rank：

x = tf.placeholder(tf.float32, [None, 10])
v = tf.Variable(tf.ones_like(x), trainable=False, validate_shape=False)
tf.layers.dense(v, 10)

ValueError: Input 0 of layer dense_5 is incompatible with the layer: its rank is undefined, but the layer requires a defined rank.

在这种情况下，虽然确切的变量形状必须是动态的，但我知道它的等级是什么。有没有办法告诉它tensorflow，所以我可以使用需要知道输入等级的操作？

Answer 1

您可以使用tf.reshape()：

执行此操作

x = tf.placeholder(tf.float32, [None, 10])
v = tf.Variable(tf.ones_like(x), trainable=False, validate_shape=False)
#tf.layers.dense(v, 10)
tf.layers.dense(tf.reshape(v,[-1,10]),10)

-1允许返回张量具有形状(?,10);即上面的输出是：

<tf.Tensor 'dense_10/BiasAdd:0' shape=(?, 10) dtype=float32>

这就是你想要的。您可以使用已知形状并切换validate_shape来验证正确的行为，如：

x = tf.placeholder(tf.float32, [5, 10])
v = tf.Variable(tf.ones_like(x), trainable=False, validate_shape=True)
tf.layers.dense(v, 10)

...返回与：

相同

x = tf.placeholder(tf.float32, [5, 10])
v = tf.Variable(tf.ones_like(x), trainable=False, validate_shape=False)
tf.layers.dense(tf.reshape(v,[5,10]),10)

# returns <tf.Tensor 'dense_8/BiasAdd:0' shape=(5, 10) dtype=float32>

在Tensorflow中确定具有未知形状的变量等级

1 个答案: