Thank you, but I could not really see how this applies to my situation.
Are you suggesting that I would require a custom nn.Module subclass which basically does the same as Sequential except has an additional for-loop in the forward to go through the third dimension?