使用pytorch将三维图像分块(patches)

问题：高光谱图像输入为5维，即(batch_size, channels, bands, height, width) ，需要将其分为多个大小为(k, k, k)的小块，但pytorch的unfold函数只支持输入是四维的情况
解决方法：使用单独的包
https://github.com/f-dangel/unfoldNd