TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.contrib.framework.with_shape

View source on GitHub

Asserts tensor has expected shape.

tf.contrib.framework.with_shape(
    expected_shape,
    tensor
)

If tensor shape and expected_shape, are fully defined, assert they match. Otherwise, add assert op that will validate the shape when tensor is evaluated, and set shape on tensor.

Args:

  • expected_shape: Expected shape to assert, as a 1D array of ints, or tensor of same.
  • tensor: Tensor whose shape we're validating.

Returns:

tensor, perhaps with a dependent assert operation.

Raises:

  • ValueError: if tensor has an invalid shape.