About the signal transform

[dmkr001]

Hi, I am a student now work on omnidirectional image processing. I am interested in this work. I have two questions about this. First I print the tensor shape after processing from the SO3conv layers, but as I know it should be like [batch size, channels, 2b, 2b] because the output of SO3conv should in the S2 domain? But what I got is [batch size, channels, 2b, 2b, 2*b]. And another question is that I want to use these layers to build an autoencoder, how can I project the signal back to the input image domain? I means from SO3 project back to S2. @jonas-koehler @mariogeiger @tscohen @AndreaPi @pochoi