Explainable representation learning of small quantum states