docs: def _ImageDimensions(images, dynamic_shape=False):
    """Returns the dimensions of an image tensor.
    Args:
      images: 4-D Tensor of shape [batch, height, width, channels]
      dynamic_shape: Whether the input image has undertermined shape. If set to
        `True`, shape information will be retrieved at run time. Default to
        `False`.
  
    Returns:
      list of integers [batch, height, width, channels]
    """
    # A simple abstraction to provide names for each dimension. This abstraction
    # should make it simpler to switch dimensions in the future (e.g. if we ever
    # want to switch height and width.)
    if dynamic_shape:
      return array_ops.unpack(array_ops.shape(images))
    else:
      return images.get_shape().as_list()