argmax function

VARP argmax(
  1. VARP x, {
  2. int? axis,
})

Implementation

VARP argmax(VARP x, {int? axis}) {
  if (axis == null) {
    return F.argMax(ravel(x), 0);
  }
  return F.argMax(x, axis);
}