Merge pull request #973 from Conchylicultor:patch-1

PiperOrigin-RevId: 355347049
google · Feb 3, 2021 · 278df51 · 278df51
2 parents 61580b9 + d292279
commit 278df51
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/docs/flax.linen.rst b/docs/flax.linen.rst
@@ -37,6 +37,7 @@ Linear modules
     Dense
     DenseGeneral
     Conv
+    ConvTranspose
     Embed
 
 

diff --git a/flax/linen/linear.py b/flax/linen/linear.py
@@ -288,7 +288,7 @@ def __call__(self, inputs: Array) -> Array:
 
 
 class ConvTranspose(Module):
-  """Convolution Module wrapping lax.conv_general_dilated.
+  """Convolution Module wrapping lax.conv_transpose.
 
   Attributes:
     features: number of convolution filters.
-Original file line number
+Diff line change
@@ Expand Up / @@ -37,6 +37,7 @@ Linear modules @@
         Dense
         DenseGeneral
         Conv
+        ConvTranspose
         Embed
@@ Expand Down @@