tf.contrib.framework.with_shape

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2020-10-01 UTC.

tf.contrib.framework.with_shape

Args

Returns

Raises