W3cubDocs

/TensorFlow Python

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

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

See the guide: Framework (contrib)

Asserts tensor has expected shape.

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.

Defined in tensorflow/contrib/framework/python/framework/tensor_util.py.

© 2017 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/framework/with_shape