Missed TensorFlow World? Check out the recap. Learn more

tf.contrib.data.assert_element_shape

tf.contrib.data.assert_element_shape(expected_shapes)

Defined in tensorflow/contrib/data/python/ops/batching.py.

Assert the shape of this Dataset.

shapes = [tf.TensorShape([16, 256]), tf.TensorShape([None, 2])]
result = dataset.apply(tf.contrib.data.assert_element_shape(shapes))
print(result.output_shapes)  # ==> "((16, 256), (<unknown>, 2))"

If dataset shapes and expected_shape, are fully defined, assert they match. Otherwise, add assert op that will validate the shapes when tensors are evaluated, and set shapes on tensors, respectively.

Note that unknown dimension in expected_shapes will be ignored.

Args:

Returns:

A Dataset transformation function, which can be passed to tf.data.Dataset.apply