On Generalization Bounds for Neural Networks with Low Rank Layers