jit.submatrix

If you want to isolate a particular subset of a matrix—such as a single column, or a single row, or really any rectangular portion of the matrix—a handy object for that purpose is jit.submatrix.

This example demonstrates how to find the maximum value in a jit.matrix column using jit.submatrix and jit.3m.

This example uses a 4-plane, 256×256 matrix for easy visualization, but it could be adapted to a 6-plane 1×1024 matrix.