NVIDIA / MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
https://nvidia.github.io/MinkowskiEngine
Other
2.47k stars 367 forks source link

Change the Dimensions used by the convolution Once one Dimension Collapses After Convolution #522

Open jrockholt3 opened 1 year ago

jrockholt3 commented 1 year ago

I am apply convolution to a 4D tensor but my tensor is only 6 elements deep in the 1st dimension. After applying the first two convolution layers, I want to reshape the sparse tensor to a 3D tensor and then use 3-D convolution on that tensor. I cannot find anything on the documentation that would allow me to easily to do this.