diff options
| author | Eric Kunze <eric.kunze@arm.com> | 2024-03-11 21:20:42 -0700 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2024-03-14 21:20:05 -0700 |
| commit | fb65bcf27eb77f593abd345aed0e884a440c827f (patch) | |
| tree | 3acd3ebecbf450efa9fd74b498645bfa15115ac8 | |
| parent | 0afe61f88ce3d2f445c5f01ae5567cb1b0b7f303 (diff) | |
| download | specification-fb65bcf27eb77f593abd345aed0e884a440c827f.tar.gz | |
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 <eric.kunze@arm.com>
Change-Id: Id05a680cd11a331ba84a8e38f9a7c77a71bcbb78
| -rw-r--r-- | chapters/tensor_ops.adoc | 3 |
1 files changed, 3 insertions, 0 deletions
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[] |