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.
|