Update image.py

ASEM000 · Aug 18, 2023 · 40438c6 · 40438c6
1 parent e87f4c7
commit 40438c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/serket/nn/image.py b/serket/nn/image.py
@@ -1225,7 +1225,7 @@ class JigSaw2D(sk.TreeClass):
           [ 5  6  7  8]
           [ 9 10 11 12]
           [13 14 15 16]]]
-        >>> print(sk.nn.JigSaw(2)(x))
+        >>> print(sk.nn.JigSaw2D(2)(x))
         [[[ 9 10  3  4]
           [13 14  7  8]
           [11 12  1  2]