def METHOD_NAME(self, tree_node_num):
    self.embedding_lookup_program = Program()
    startup = Program()
    with paddle.static.framework.program_guard(
            self.embedding_lookup_program, startup):
        all_nodes = paddle.static.data(
            name="all_nodes",
            shape=[-1, 1],
            dtype="int64",
            lod_level=1, )
        output = paddle.static.nn.embedding(
            input=all_nodes,
            is_sparse=True,
            size=[tree_node_num, self.node_emb_size],
            param_attr=paddle.ParamAttr(
                name="tdm.bw_emb.weight",
                initializer=paddle.initializer.UniformInitializer()))
        self.embedding_fetch_var_names = [output.name]
        self.exe.run(startup)