(LearnerND) use direct neighbours in loss

same as #119 (closed) but now for R^N \to R^N