From fe0672a9d24d81dcf85a3ce48ce60157d4a86fdf Mon Sep 17 00:00:00 2001 From: John Mai Date: Sun, 15 Jun 2025 17:33:58 +0800 Subject: [PATCH] docs: Update documentation to include ReLUSquared activation function --- docs/src/python/nn/functions.rst | 1 + docs/src/python/nn/layers.rst | 1 + 2 files changed, 2 insertions(+) diff --git a/docs/src/python/nn/functions.rst b/docs/src/python/nn/functions.rst index 9b6cd9f62..325a24ef2 100644 --- 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 diff --git a/docs/src/python/nn/layers.rst b/docs/src/python/nn/layers.rst index 4eb14b088..218f7f457 100644 --- 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