original_docstring: visit function and add kl_loss computation.
docstring: visit function and add kl_loss computation.
docstring_tokens: visit function and add kl_loss computation .
code: def visit_FunctionDef(self, node):
          """visit function and add kl_loss computation."""
          self.generic_visit(node)
          if node.name == 'cal_kl_loss':
              for i in range(self.layer_count):
                  func = ast.Assign(targets=[ast.Name(id='loss', ctx=ast.Store())],
                                    value=ast.BinOp(left=ast.Name(id='loss', ctx=ast.Load()), op=ast.Add(),
                                                    right=ast.Call(func=ast.Name(id='self.kl_loss' + '[' + str(i) + ']',
                                                                                 ctx=ast.Load()),
                                                                   args=[], keywords=[])))
                  node.body.insert(-1, func)
          return node
code_tokens: def visit_FunctionDef ( self , node ) : self . generic_visit ( node ) if node . name == 'cal_kl_loss' : for i in range ( self . layer_count ) : func = ast . Assign ( targets = [ ast . Name ( id = 'loss' , ctx = ast . Store ( ) ) ] , value = ast . BinOp ( left = ast . Name ( id = 'loss' , ctx = ast . Load ( ) ) , op = ast . Add ( ) , right = ast . Call ( func = ast . Name ( id = 'self.kl_loss' + '[' + str ( i ) + ']' , ctx = ast . Load ( ) ) , args = [ ] , keywords = [ ] ) ) ) node . body . insert ( - 1 , func ) return node
short_docstring: visit function and add kl_loss computation.
short_docstring_tokens: visit function and add kl_loss computation .
comment: """visit function and add kl_loss computation."""