assert_shape_equal

tfsnippet.ops.assert_shape_equal(x, y, message=None, name=None)

Assert the shape of x equals to y.

Parameters:
  • x – A tensor.
  • y – Another tensor, to compare with x.
  • message – Message to display when assertion failed.
  • name (str) – Default name of the name scope. If not specified, generate one according to the method name.
Returns:

The TensorFlow assertion operation,

or None if can be statically asserted.

Return type:

tf.Operation or None