I am trying to use channel-wise fully-connected layer which was introduced in paper “Context Encoders: Feature Learning by Inpainting”,  however I have no idea on how to implement this in pytorch. if there  someone can give me help, thanks a lot !