def reshape_for_linear(images):
    """Reshape the images for the linear model
    Our linear model requires that the images be reshaped as a 1D tensor
    """
    n_images, n_rgb, img_height, img_width = images.shape
    return images.reshape(n_images, n_rgb * img_height * img_width)