private void setIndexes(BaseSparseNDArrayCOO coo, boolean oneBased) { int incr = oneBased ? 1 : 0; int[] idx = coo.getIncludedIndices().asInt(); int[] rows = new int[nnz]; int[] cols = new int[nnz]; for (int i = 0; i < nnz; i++) { rows[i] = idx[i * 2] + incr; cols[i] = idx[(i * 2) + 1] + incr; } rowInd = Nd4j.createBuffer(rows); colInd = Nd4j.createBuffer(cols); }