Following the tutorial:
https://pytorch.org/tutorials/intermediate/spatial_transformer_tutorial.html 5
What would I need to do to adapt the STN network to work with specific B=batch size, H=image height, W=image width, C= channels? I assume B is handled by .view(-1, …) but what about the remaining H, W, C values?
Thank you and my apologies if this is a bit of a beginner question!