merge GeneralLinear to Linear, remove FNN (#84)

ASEM000 · Nov 28, 2023 · 3efda52 · 3efda52
1 parent aedf891
commit 3efda52
Show file tree

Hide file tree

Showing 6 changed files with 60 additions and 431 deletions.
diff --git a/docs/API/linear.rst b/docs/API/linear.rst
@@ -4,8 +4,5 @@ Linear
 
 .. autoclass:: Linear
 .. autoclass:: Identity
-.. autoclass:: GeneralLinear
 .. autoclass:: Embedding
-
-.. autoclass:: FNN
 .. autoclass:: MLP
diff --git a/serket/_src/nn/activation.py b/serket/_src/nn/activation.py
@@ -561,9 +561,6 @@ def def_act_entry(key: str, act: ActivationFunctionType) -> None:
         ...    def __call__(self, x):
         ...        return x * self.my_param
         >>> sk.def_act_entry("my_act", MyTrainableActivation())
-        >>> x = jnp.ones((1, 1))
-        >>> sk.nn.FNN([1, 1, 1], act="my_act", weight_init="ones", bias_init=None, key=jr.PRNGKey(0))(x)
-        Array([[10.]], dtype=float32)
     """
     if key in act_map:
         raise ValueError(f"`init_key` {key=} already registered")