Could you explain the reason for this permutation and print the shape of x before permuting it?
I assume the permute op is related to the XNOR model, but I’m unfortunately not deeply familiar with it.