Prediction and generalisation over directed actions by grid cells