Revert "FIX Update cifar_small_sample.py" (#1046)

This reverts commit b8b1d53.

examples/2d/cifar_small_sample.py

@@ -9,7 +9,7 @@
 import torch.optim
 from torchvision import datasets, transforms
 import torch.nn.functional as F
-from kymatio.torch import Scattering2D
+from kymatio import Scattering2D
 import torch
 import argparse
 import kymatio.datasets as scattering_datasets