From ca8a4169a8270b46875c88c89e183e0c1c4efdca Mon Sep 17 00:00:00 2001
From: Fatih Cetinkaya <965295+devfacet@users.noreply.github.com>
Date: Fri, 5 Jul 2024 18:29:27 -0400
Subject: [PATCH] Update comments

---
 include/nn_activation.h | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/include/nn_activation.h b/include/nn_activation.h
index 175ca69..798a4d8 100644
--- a/include/nn_activation.h
+++ b/include/nn_activation.h
@@ -50,6 +50,8 @@ typedef struct {
  *
  * @param type The type of activation function (scalar or tensor).
  * @param func The activation function.
+ *
+ * @return The activation function.
  */
 NNActFunc nn_act_func_init(NNActFuncType type, void *func);