Holen Sie sich die Größe einer variable batch-dimension

Wenn x hat eine variable batch-Größe, der einzige Weg, um die tatsächliche Form ist die Verwendung der tf.Form() Betreiber. Dieser operator gibt einen symbolischen Wert in einer tf.Tensor, so kann es verwendet werden als Eingang für andere TensorFlow Operationen, sondern um eine konkrete Python-Wert für die Form, die Sie brauchen, um es zu übergeben, um Session.run().

x = tf.placeholder(..., shape=[None, ...])
batch_size = tf.shape(x)[0]  # Returns a scalar `tf.Tensor`

print x.get_shape()[0]  # ==> "?"

# You can use `batch_size` as an argument to other operators.
some_other_tensor = ...
some_other_tensor_reshaped = tf.reshape(some_other_tensor, [batch_size, 32, 32])

# To get the value, however, you need to call `Session.run()`.
sess = tf.Session()
x_val = np.random.rand(37, 100, 100)
batch_size_val = sess.run(batch_size, {x: x_val})
print x_val  # ==> "37"