Not that I know of. You'd need to use codebox and a for loop over the input matrix to do the neighbor calculations. Using a 1D matrix would make things simpler.

Thanks, yes I was also thinking of a 1D matrix, I once built the algorithm in processing and there I also used a 1D array.

But I'm new to gen so I was wandering if it could be done.

A 1D matrix is only possible in jit.gen right? Because I would like to run it on the GPU.

jit.gl.pix runs on the GPU. jit.gen and jit.pix on the CPU. Here's a jit.gen version:

wow, super thanks! This helps me a lot.