tf.contrib.framework.with_same_shape

tf.contrib.framework.with_same_shape(
    expected_tensor,
    tensor
)

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

See the guide: Framework (contrib)

Assert tensors are the same shape, from the same graph.

Args:

• expected_tensor: Tensor with expected shape.
• tensor: Tensor of actual values.

Returns:

The original tensor argument, possibly with assert ops added.