Merge pull request #39 from DarrenZhang01/updated_changes

Revise the typo from `dot_general_dilated` to `conv_general_dilated`.
google · Jun 29, 2020 · 522bfb8 · 522bfb8
2 parents 46edb0e + e2c69a0
commit 522bfb8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neural_tangents/stax.py b/neural_tangents/stax.py
@@ -483,7 +483,7 @@ def GeneralConv(
 
   Args:
     dimension_numbers: Specifies which axes should be convolved over. Should
-      match the specification in `jax.lax.dot_general_dilated`.
+      match the specification in `jax.lax.conv_general_dilated`.
     out_chan: The number of output channels / features of the
       convolution. This is ignored in by the `kernel_fn` in NTK
       parameterization.