def get_spatial_slice(self, num):
img = self.U[...,num].reshape((self._n_y, self._n_x))
return _np.real(img)
# Used to return complex, but the SVD of complex numbers tends to
# shove everything in U into the real component
# if _np.iscomplexobj(img):
# if self._use_imag:
# return _np.imag(img)
# else:
# return _np.real(img)
# else:
# return img
评论列表
文章目录