Commit 716d5299 authored by Atreya Majumdar's avatar Atreya Majumdar
Browse files

Added LayerNorm to docs

parent 2865bec7

docs/source/api_reference/layers.rst

+3 −0
Original line number Diff line number Diff line
@@ -112,4 +112,7 @@ another tensor. DeepChem maintains an extensive collection of layers which perfo 
.. autoclass:: deepchem.models.layers.SetGather

  :members:



.. autoclass:: deepchem.models.torch_models.layers.LayerNorm

  :members:



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

CRA Git | Maintained and supported by SUSTech CRA and CCSE