About the inverse operation for cifar-dataset

[momo1986]

Hello,

Thanks a lot for your generous sharing.

I want to use your proposed structure in some adversarial training framework.

My problem is that:

When batch size is 64, the previous data has the size of 64332*32

For J=1, after scattering, the matrix size is 64391616.
For J=2, after scattering, the matrix size is 6431788.
For J=3, after scattering, the matrix size is 6432544.
For J=4, after scattering, the matrix size is 6433322.

Looks like the tensor-shape for third dimesion is 1+J*8, the fourth and fifth dimension is 32/2^j.

I am not sure, is there a inverse operation API that I can restore the matrix after scattering to the original matrix, e.g., Tensor with size 64332*32.

It would be honor if you could answer the questions.
Thanks & regards!
Momo