def forward(self, *args):
        """Creates a view (copy) of the input with given shape."""
        assert len(args) == 2
        x = args[0]
        shape = args[1]
        return self.op(x, shape)
Creates a view (copy) of the input with given shape.
python/aitemplate/frontend/nn/view_ops.py