prototorch/docs/source/api.rst

47 lines
1.0 KiB
ReStructuredText
Raw Normal View History

.. ProtoTorch API Reference
2021-04-14 17:20:08 +00:00
ProtoTorch API Reference
2021-04-14 17:20:08 +00:00
======================================
Datasets
--------------------------------------
.. automodule:: prototorch.datasets
:members:
:undoc-members:
Functions
--------------------------------------
2021-04-21 12:55:54 +00:00
**Dimensions:**
- :math:`B` ... Batch size
- :math:`P` ... Number of prototypes
- :math:`n_x` ... Data dimension for vectorial data
- :math:`n_w` ... Data dimension for vectorial prototypes
Activations
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: prototorch.functions.activations
:members:
:exclude-members: register_activation, get_activation
:undoc-members:
Distances
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: prototorch.functions.distances
2021-04-14 17:20:08 +00:00
:members:
2021-04-21 12:55:54 +00:00
:exclude-members: sed
2021-04-14 17:20:08 +00:00
:undoc-members:
Modules
--------------------------------------
.. automodule:: prototorch.modules
:members:
:undoc-members:
Utilities
--------------------------------------
.. automodule:: prototorch.utils
:members:
:undoc-members: