用神经网络模拟原子