Equality reasoning in neural networks is modulated by learning richness