Method forward

imperative/python/megengine/module/vision.py:820–830

(self, inp)

        self.imode = imode

    def forward(self, inp):
        assert isinstance(
            inp, Tensor
        ), "expected input is megengine.Tensor, got {}".format(type(inp))
        if self.format not in ["NCHW", "NHWC"]:
            raise RuntimeError(
                "expect input Tensor format to be NCHW or NHWC, got format {}".format(
                    self.format
                )
            )
        return resize(inp, self.size, self.format, self.imode)


class Rot90(Module):