docs: Update documentation to include ReLUSquared activation function

2025-06-24 01:17:26 +08:00 · 2025-06-15 17:33:58 +08:00 · 2025-06-15 17:33:58 +08:00 · fe0672a9d2
commit fe0672a9d2
parent cbd353bf73
2 changed files with 2 additions and 0 deletions
--- a/docs/src/python/nn/functions.rst
+++ b/docs/src/python/nn/functions.rst
@ -28,6 +28,7 @@ simple functions.
   prelu
   relu
   relu6
+   relu_squared
   selu
   sigmoid
   silu
--- a/docs/src/python/nn/layers.rst
+++ b/docs/src/python/nn/layers.rst
@ -51,6 +51,7 @@ Layers
   RMSNorm
   ReLU
   ReLU6
+   ReLUSquared
   RNN
   RoPE
   SELU