@lerezell Thank you for your answer. My input is a video. So, I want to see one of its frame. But my question is when I call that method, do I just call it as
                self.res1, self. outAfterMask, mask = residal_unit(128, 128)

Later,
                              output=model(input)

How do I get that MyTensor?