Update comments

devfacet · Jul 5, 2024 · ca8a416 · ca8a416
1 parent b7b6215
commit ca8a416
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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);