def assert_shape_match(shape1, shape2):
  """Ensure the shape1 match the pattern given by shape2.

  Ex:
    assert_shape_match((64, 64, 3), (None, None, 3))

  Args:
    shape1 (tuple): Static shape
    shape2 (tuple): Dynamic shape (can contain None)
  """
  shape1 = tf.TensorShape(shape1)
  shape2 = tf.TensorShape(shape2)
  if shape1.ndims is None or shape2.ndims is None:
    raise ValueError('Shapes must have known rank. Got %s and %s.' %
                     (shape1.ndims, shape2.ndims))
  shape1.assert_same_rank(shape2)
  shape1.assert_is_compatible_with(shape2)
def assert_shape_match ( shape1 , shape2 ) : shape1 = tf . TensorShape ( shape1 ) shape2 = tf . TensorShape ( shape2 ) if shape1 . ndims is None or shape2 . ndims is None : raise ValueError ( 'Shapes must have known rank. Got %s and %s.' % ( shape1 . ndims , shape2 . ndims ) ) shape1 . assert_same_rank ( shape2 ) shape1 . assert_is_compatible_with ( shape2 )
Ensure the shape1 match the pattern given by shape2.

  Ex:
    assert_shape_match((64, 64, 3), (None, None, 3))

  Args:
    shape1 (tuple): Static shape
    shape2 (tuple): Dynamic shape (can contain None)