diff options
-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[] |