Currently, only 4-D input tensors (batched image-like tensors) are supported by unfold and fold
I assume the size is (N, C, H, W)
a set of 3D medical images are 5D tensors:  (N, C, D, H, W)
When will unfold and fold support 5D tensors?