Adding attention to the neural ordinary differential equation for spatio-temporal prediction