6.4.5 自定义的Pytree节点