Basically what I need to do is I have a tensor of shape (b, c, h, w)
I have to process each of the (h, w) in the tensor separately with a custom layer that only takes bidimensional input, so I just iterated over the b and c dimensions and appended each result.
Isn’t there any other way to do this with a pytorch function?