@thi.ng/tensors/find.js

0D/1D/2D/3D/4D tensors with extensible polymorphic operations and customizable storage

const findIndex = (a, pred, start = 0, end = a.length) => {
  const {
    offset: oa,
    stride: [ta],
    data: adata
  } = a;
  for (let i = start; i < end; i++) {
    if (pred(adata[oa + i * ta])) return i;
  }
  return -1;
};
export {
  findIndex
};