class_docstr: Attention net.  See file docstring.
class_docstr_tok: ["Attention","net",".","See","file","docstring","."]
code: def _normalize_body_output(self, body_out):
      if len(body_out) == 2:
          output, losses = body_out
          if not isinstance(losses, dict):
              losses = {"extra": tf.reduce_mean(losses)}
      else:
          output = body_out
          losses = {"extra": 0.0}
  
      return output, losses
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import six
  from tensor2tensor.layers import common_attention
  from tensor2tensor.layers import common_layers
  from tensor2tensor.models import transformer
  from tensor2tensor.utils import registry
  import tensorflow.compat.v1
  from tensorflow.compat.v1 import estimator
label_desc: Use image node_id 6 for calling the NeuralAssistant obj's underlying member method code with example usage: obj._normalize_body_output(body_out) and returns: output,  losses