Guaranteed Conservationof Momentum for Learning Particle-based Fluid Dynamics