Hi,
I don’t think there is any plan to support generalised versions of fold and unfold.
You have Conv3D if you need to do convolutions though.
@fmassa migh be more familiar about how easy/hard it would be to generalise fold and unfold?