From fb65bcf27eb77f593abd345aed0e884a440c827f Mon Sep 17 00:00:00 2001 From: Eric Kunze Date: Mon, 11 Mar 2024 21:20:42 -0700 Subject: Tighten ARGMAX definition Define behavior if multiple values are tied for the maximum. Define behavior if NaN values are encountered. Signed-off-by: Eric Kunze Change-Id: Id05a680cd11a331ba84a8e38f9a7c77a71bcbb78 --- chapters/tensor_ops.adoc | 3 +++ 1 file changed, 3 insertions(+) diff --git a/chapters/tensor_ops.adoc b/chapters/tensor_ops.adoc index 3de5150..a36a7f1 100644 --- a/chapters/tensor_ops.adoc +++ b/chapters/tensor_ops.adoc @@ -12,6 +12,9 @@ ==== ARGMAX This returns the index with the largest value across the given axis of the input tensor. +If multiple locations have equal values, returns the first match along the search axis. +NaN values always compare as greater than non-NaN values. +If an entire axis consists of NaN, the last location must be returned. include::{generated}/operators/ARGMAX.adoc[] -- cgit v1.2.1