Commit 9e397b2b authored by Atreya Majumdar's avatar Atreya Majumdar
Browse files

Added to layers.rst

parent d634d11a
Loading
Loading
Loading
Loading
+2 −8
Original line number Diff line number Diff line
@@ -125,16 +125,10 @@ Torch Layers
.. autoclass:: deepchem.models.torch_models.layers.ScaleNorm
  :members:

.. autoclass:: deepchem.models.torch_models.layers.MATEncoderLayer
.. autoclass:: deepchem.models.torch_models.layers.MATEmbedding
  :members:

.. autoclass:: deepchem.models.torch_models.layers.MultiHeadedMATAttention
  :members:

.. autoclass:: deepchem.models.torch_models.layers.SublayerConnection
  :members:

.. autoclass:: deepchem.models.torch_models.layers.PositionwiseFeedForward
.. autoclass:: deepchem.models.torch_models.layers.MATGenerator
  :members:

.. autofunction:: deepchem.models.layers.cosine_dist