Thanks for sharing your code which helps a lot!
However, I met some problems when I trying to modifying the in_channel.
For example, I set
input_dim = [5, 7, 7]
in_channels = 2
out_channels = 4
kernel_size = (4, 6, 6)
padding = (1, 1, 1)
stride = (1, 1, 1)
then it shows
RuntimeError: size mismatch, m1: [4 x 288], m2: [144 x 128] at /Users/distiller/project/conda/conda-bld/pytorch_1570710797334/work/aten/src/TH/generic/THTensorMath.cpp:197
then I set
unfolded_input_tensor = unfolded_input_tensor.reshape(-1, in_channels * kernel_size[0] * kernel_size[1] * kernel_size[2])
since the kernel shape is out_c*in_c*k1*k2*k3
Eventhough I pass the error, the final result is tensor(False)
I would like to know how to unroll Conv3d in multi-input channels.
Thanks for your time to answer the question!