def __getitem__(self, slices):
data = self._data[slices]
toworld = subgrid_affine(self._toworld, slices)
res = GridTransform((data.shape[:-2], toworld), data, self._affine)
res._param[:] = self._param[:]
res._free_param_idx = self._free_param_idx
return res
def __getitem__(self, slices):
toworld = subgrid_affine(self._toworld, slices)
fake_data = np.ones(self._shape, dtype='bool')[slices]
res = SplineTransform((fake_data.shape, toworld),
self._control_points, self._sigma,
grid_coords=False, affine=self._affine)
res._param[:] = self._param[:]
res._free_param_idx = self._free_param_idx
return res