import paddle



class Model(paddle.nn.Layer):
    def __init__(self):
        super().__init__()
        self.u = paddle.to_tensor(1.0)
        self.v = paddle.to_tensor(2.0)
        self.w = paddle.to_tensor(3.0)
        self.p = paddle.to_tensor(4.0)

    def forward(self):
        return {
            "u": self.u,
            "v": self.v,
            "w": self.w,
            "p": self.p,
        }


model = Model()

dy_output = model()
# print(list(dy_output.values()))

st_model = paddle.jit.to_static(model)
st_output = st_model()
# print(list(st_output.values()))

paddle.jit.save(st_model, "dy2st")
jit_load_model = paddle.jit.load("dy2st")
jit_output = jit_load_model()
print(list(jit_output)) # <--- here