diff options
Diffstat (limited to 'candle-pyo3/py_src/candle/nn/__init__.py')
| -rw-r--r-- | candle-pyo3/py_src/candle/nn/__init__.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/candle-pyo3/py_src/candle/nn/__init__.py b/candle-pyo3/py_src/candle/nn/__init__.py index b8c5cfb7..8da0e8aa 100644 --- a/candle-pyo3/py_src/candle/nn/__init__.py +++ b/candle-pyo3/py_src/candle/nn/__init__.py @@ -1,5 +1,5 @@ -# Generated content DO NOT EDIT -from .. import nn - -silu = nn.silu -softmax = nn.softmax +from .module import Module +from .container import Sequential, ModuleList, ModuleDict +from .sparse import Embedding +from .normalization import LayerNorm +from .linear import Linear |