code: 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)
docs: 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)