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

Question

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

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.

在这种情况下，尽管确切的变量形状必须是动态的，但我知道其排名将是多少。有什么方法可以将其告知张量流，以便我可以使用需要知道输入等级的操作吗？

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>