Learning low-dimensional generalizable natural features from retina using a U-net