利用Python装饰器（decorator）来改善Tensorflow代码的结构