def create_example(images):
  """Creates training example of inputs and label from `images`."""
  images.shape.assert_is_compatible_with([None, None, 9])

  inputs = {
      'frame_0': images[Ellipsis, 0:3],
      'frame_1': images[Ellipsis, 3:6],
  }
  label = images[Ellipsis, 6:9]
  return inputs, label