hello,
so im trying to visualize the ouput of my spatial transformer component and when i try to apply a transpose to the output of the stn which is of shape (1, 50, 22, 22) by applying a transpose of (1,2,0) i get an axes dont match error. im trying to follow this demo:http://pytorch.org/tutorials/intermediate/spatial_transformer_tutorial.html 2 as to how im going to visualize the stn.
Any help is greatly appreciated