for yy in range(num_steps[0]):
        for xx in range(num_steps[1]):
            start_y = min(yy * step[0], img_height - window[0])
            start_x = min(xx * step[1], img_width - window[1])
            stop_y = start_y + window[0]
            stop_x = start_x + window[1]

class NCYXQuilt(object):
    .
    .
    def unstitch(self, tensor):
            """
            Unstich a single tensor.
    
            Parameters
            ----------
            tensor
    
            Returns
            -------
            A patched tensor
            """
            N, C, Y, X = tensor.shape
            result = []
            list_of_boundaries = self.get_patches_boundaries()
            num_patches = len(list_of_boundaries)
    
            for n in range(N):
                tmp = tensor[n, ...]
                for ii in num_patches:
                        start_x, start_y, stop_x, stop_y = list_of_boundaries[ii]
                        patch = tmp[:, start_y:stop_y, start_x:stop_x]
                        result.append(patch)
            result = einops.rearrange(result, "M C Y X -> M C Y X")
            return result
    
    def get_patch_boundaries(self, xx, yy):
            """
            Get the path boundaries based on the location of the patch 
    
            Parameters
            ----------
            List
    
            Returns
            -------
            List of boundary locations of patch
            """
            start_y = min(yy * self.step[0], self.Y - self.window[0])
            start_x = min(xx * self.step[1], self.X - self.window[1])
            stop_y = start_y + self.window[0]
            stop_x = start_x + self.window[1]
            return start_x, start_y, stop_x, stop_y
    
    def get_patches_boundaries(self):
            """
            Should make use of get_patch_boundaries to return a list of boundaries
            """
    .....