GridSplit has two separate implementations for torch.Tensor and numpy.ndarray. Although the torch implementation using unfold is specific to torch, both should be able to use the numpy implementation using as_strided. Thus, it would be nice to compare the performance of these two implementation and if as_strided is better, we can unify the implementation. Other factor to consider would be ability to expand to tensors with three spatial dimensions.