diff options
Diffstat (limited to 'chapters/tensor_ops.adoc')
-rw-r--r-- | chapters/tensor_ops.adoc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/chapters/tensor_ops.adoc b/chapters/tensor_ops.adoc index cb268fa..508b680 100644 --- a/chapters/tensor_ops.adoc +++ b/chapters/tensor_ops.adoc @@ -17,7 +17,7 @@ include::{generated}/operators/ARGMAX.adoc[] [source,c++] ---- -ERROR_IF(axis < 0 || axis >= rank(shape1) || rank(shape1) > 4); +ERROR_IF(axis < 0 || axis >= rank(shape1)); if (axis == 0) { left_shape = []; } else { |