Merge pull request #72 from Tensor46/develop

+ docs
Tensor46 · Feb 21, 2021 · 67617d3 · 67617d3
2 parents 168f241 + e8264da
commit 67617d3
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -2,7 +2,6 @@
 __pycache__/
 *.py[cod]
 *$py.class
-YellowFIN.py
 test*
 .idea/
 .DS_Store

diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -20,6 +20,7 @@ TensorMONK is a computer vision library for `PyTorch <https://pytorch.org/>`_.
 
    modules/activations
    modules/detection
+   modules/layers
 
 
 Indices and Tables

diff --git a/docs/source/modules/layers.rst b/docs/source/modules/layers.rst
@@ -0,0 +1,40 @@
+tensormonk.layers
+==================
+
+Attention, convolution, linear and vision layers.
+
+.. contents:: Contents
+    :local:
+
+CondConv2d
+-----------
+
+.. autoclass:: tensormonk.layers.CondConv2d
+    :members:
+    :exclude-members: forward
+
+Attention Layers
+-----------------
+
+All attention layers.
+
+LocalAttention
+^^^^^^^^^^^^^^^
+
+.. autoclass:: tensormonk.layers.LocalAttention
+    :members:
+    :exclude-members: forward
+
+SelfAttention
+^^^^^^^^^^^^^^
+
+.. autoclass:: tensormonk.layers.SelfAttention
+    :members:
+    :exclude-members: forward
+
+
+LucasKanade
+------------
+.. autoclass:: tensormonk.layers.LucasKanade
+    :members:
+    :exclude-members: extract_patches, gaussian