Simple and fast digital network (network dolls in the network)

import paddle
class Tree_net(paddle.nn.Layer):
    def __init__(self, hidden_dim=100):
        super(Tree_net, self