Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
121 lines (85 sloc) 1.7 KB

pixyz.flows (Flow layers)

.. automodule:: pixyz.flows
.. currentmodule:: pixyz.flows

Flow

.. autoclass:: Flow
    :members:
    :undoc-members:

.. autoclass:: FlowList
    :members:
    :undoc-members:

Normalizing flow

PlanarFlow

.. autoclass:: PlanarFlow
    :members:
    :undoc-members:


Coupling layer

AffineCoupling

.. autoclass:: AffineCoupling
    :members:
    :undoc-members:

Invertible layer

ChannelConv

.. autoclass:: ChannelConv
    :members:
    :undoc-members:

Operation layer

Squeeze

.. autoclass:: Squeeze
    :members:
    :undoc-members:

Unsqueeze

.. autoclass:: Unsqueeze
    :members:
    :undoc-members:

Permutation

.. autoclass:: Permutation
    :members:
    :undoc-members:

Shuffle

.. autoclass:: Shuffle
    :members:
    :undoc-members:

Reverse

.. autoclass:: Reverse
    :members:
    :undoc-members:

Flatten

.. autoclass:: Flatten
    :members:
    :undoc-members:

BatchNorm1d

.. autoclass:: BatchNorm1d
    :members:
    :undoc-members:

BatchNorm2d

.. autoclass:: BatchNorm2d
    :members:
    :undoc-members:

ActNorm2d

.. autoclass:: ActNorm2d
    :members:
    :undoc-members:

Preprocess

.. autoclass:: Preprocess
    :members:
    :undoc-members:
You can’t perform that action at this time.