fix for gpu

smribet · smribet · commit dfd24258380d · 2025-06-03T06:23:18.000-07:00
diff --git a/py4DSTEM/tomography/tomography.py b/py4DSTEM/tomography/tomography.py
@@ -1893,6 +1893,8 @@ def _back(
             np.sum(normalize_diff == 1)
         )
         keep_diff[ind_false] = False
+        keep_diff = xp.asarray(keep_diff)
+        normalize_diff = xp.asarray(normalize_diff)
 
         update_reshaped = (
             (xp.tile(xp.repeat(update, 2, axis=1)[:, keep_diff] / normalize_diff, 4))[

Original file line number	Diff line number	Diff line change
`@@ -1893,6 +1893,8 @@ def _back(`
`1893`	`1893`	`np.sum(normalize_diff == 1)`
`1894`	`1894`	`)`
`1895`	`1895`	`keep_diff[ind_false] = False`
	`1896`	`+ keep_diff = xp.asarray(keep_diff)`
	`1897`	`+ normalize_diff = xp.asarray(normalize_diff)`
`1896`	`1898`
`1897`	`1899`	`update_reshaped = (`
`1898`	`1900`	`(xp.tile(xp.repeat(update, 2, axis=1)[:, keep_diff] / normalize_diff, 4))[`