docs: Update documentation to include ReLUSquared activation function

2025-12-16 01:49:05 +08:00 · 2025-06-15 17:33:58 +08:00
parent cbd353bf73
commit fe0672a9d2
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