#####################################
PyTorch Extension
#####################################

.. automodule:: dnois.torch

**********************************
Mathematics
**********************************
.. autosummary::
    :toctree: ../generated/torch/mathematics

    abs2
    expi
    polynomial

*********************************
Functionalities for module
*********************************
.. autosummary::
    :toctree: ../generated/torch/module

    WrapperModule
    ParamTransformModule
    Transform

*********************************
Tensor operation
*********************************
.. autosummary::
    :toctree: ../generated/torch/tensor

    as1d
    broadcastable