mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-26 02:33:21 +08:00
58 lines
1.0 KiB
ReStructuredText
58 lines
1.0 KiB
ReStructuredText
mlx.nn.Module
|
|
=============
|
|
|
|
.. currentmodule:: mlx.nn
|
|
|
|
.. autoclass:: Module
|
|
|
|
|
|
.. automethod:: __init__
|
|
|
|
|
|
.. rubric:: Methods
|
|
|
|
.. autosummary::
|
|
|
|
~Module.__init__
|
|
~Module.apply
|
|
~Module.apply_to_modules
|
|
~Module.children
|
|
~Module.clear
|
|
~Module.copy
|
|
~Module.eval
|
|
~Module.filter_and_map
|
|
~Module.freeze
|
|
~Module.fromkeys
|
|
~Module.get
|
|
~Module.is_module
|
|
~Module.items
|
|
~Module.keys
|
|
~Module.leaf_modules
|
|
~Module.load_weights
|
|
~Module.modules
|
|
~Module.named_modules
|
|
~Module.parameters
|
|
~Module.pop
|
|
~Module.popitem
|
|
~Module.save_weights
|
|
~Module.setdefault
|
|
~Module.train
|
|
~Module.trainable_parameter_filter
|
|
~Module.trainable_parameters
|
|
~Module.unfreeze
|
|
~Module.update
|
|
~Module.valid_child_filter
|
|
~Module.valid_parameter_filter
|
|
~Module.values
|
|
|
|
|
|
|
|
|
|
|
|
.. rubric:: Attributes
|
|
|
|
.. autosummary::
|
|
|
|
~Module.training
|
|
|
|
|