From 586971e678584cb8721be1532e4455ed75dee78a Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=98=A4=ED=98=95=EC=84=9D/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Staff=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Tue, 20 Nov 2018 19:09:14 +0900 Subject: [PATCH] Fix comment for argmax_ex in NeuralNetworksEx.h (#3606) Fix incorrect comment for output of argmax_ex in NeuralNetworksEx.h - Output rank should be same rank with input - Reduced axes are retained with length 1 Signed-off-by: Hyeongseok Oh --- include/NeuralNetworksEx.h | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/include/NeuralNetworksEx.h b/include/NeuralNetworksEx.h index 4b6e93b..6973600 100644 --- a/include/NeuralNetworksEx.h +++ b/include/NeuralNetworksEx.h @@ -234,7 +234,8 @@ typedef enum { /** * Find index with the largest value across axes of a input tensor. * - * Reduces the input tensor along the given dimensions to reduce. + * Reduces the input tensor along the given dimensions to reduce. The reduced + * dimensions are retained with length 1. * * Supported tensor {@link OperandCode}: * * {@link ANEURALNETWORKS_TENSOR_FLOAT32} @@ -247,7 +248,7 @@ typedef enum { * * Outputs: * * 0: A output tensor of {@link ANEURALNETWORKS_TENSOR_INT32}. - * The rank of output tensor should be rank(input_tensor)-1. + * The rank of output tensor should be same rank of input0. */ ANEURALNETWORKS_ARGMAX_EX = 50015, -- 2.7.4