fixing gaussian filter

Author: cophus

py4DSTEM/tomography/tomography.py

@@ -2056,9 +2056,16 @@ def _constraints(
 
             obj_6D = copy_to_device(self.object_6D, device="cpu")
 
-            obj_6D = gaussian_filter(
-                obj_6D, real_space_gaussian_filter, axes=(0, 1, 2)
-            )  # axes only supported in cpu
+            # obj_6D = gaussian_filter(
+            #     obj_6D, real_space_gaussian_filter, axes=(0, 1, 2)
+            # )  # axes only supported in cpu
+            for i in range(obj_6D.shape[3]):
+                for j in range(obj_6D.shape[4]):
+                    for k in range(obj_6D.shape[5]):
+                        obj_6D[:, :, :, i, j, k] = gaussian_filter(
+                            obj_6D[:, :, :, i, j, k], 
+                            sigma=real_space_gaussian_filter
+                        )
 
             self._object = copy_to_device(
                 obj_6D.reshape((s[0], s[1] * s[2], s[3] * s[4] * s[5])), device=storage