argwhere function

VARP argwhere(
  1. VARP x
)

Implementation

VARP argwhere(VARP x) {
  final mask = F.notEqual(x, F.scalar(0, dtype: x.dtype));
  return F.where(mask);
}