(self, inp)
| 876 | self.interp_mode = interp_mode |
| 877 | |
| 878 | def forward(self, inp): |
| 879 | assert isinstance( |
| 880 | inp, Tensor |
| 881 | ), "expected input is megengine.Tensor, got {}".format(type(inp)) |
| 882 | return rotate(inp, self.angle, self.format, self.interp_mode) |
| 883 | |
| 884 | |
| 885 | class Remap(Module): |