dirac basis comments

Author: chris-langfield

src/aspire/basis/dirac.py

@@ -67,16 +67,14 @@ def _evaluate_t(self, x):
              `self.count` and whose remaining dimensions correspond to
              higher dimensions of `v`.
         """
-        import pdb
-
-        pdb.set_trace()
+        # flip indexing axis to the end of the shape, i.e. (n, L, L) -> (L, L, n)
         x = m_reshape(x.asnumpy(), new_shape=self.ndim * (self.nres,) + (x.shape[0],))
         x, sz_roll = unroll_dim(x, self.ndim + 1)
         x = m_reshape(x, new_shape=(self._sz_prod,) + x.shape[self.ndim :])
         v = np.zeros(shape=(self.count,) + x.shape[1:], dtype=self.dtype)
         v = x[self._mask, ...]
         v = roll_dim(v, sz_roll)
-
+        # returns a (_sz_prod,) array for one image, (_sz_prod, n) for n images
         return np.squeeze(v)
 
     def expand(self, x):