图神经网络的微分物理模拟 这是用PyTorch编写的“的实现。 目前，基本的2D模拟效果很好。 我没有时间训练模型超过40万个时间段，因此当前结果没有像将网络训练到20M个时间段时那样完美。 在给出初始位置和五个先前速度后，以下是网络预测的一些示例： 如您所见，到目前为止，网络还没有完全学会尊重边界，并且由于某些原因，远离其他粒子的粒子会以奇怪的方向飞走（这可能是由于网络对没有粒子的粒子缺乏经验）边缘连接到他们）。 但是，它具有正确的想法，并且大多数模拟看起来都是合理的。 我预计该网络可能