parameters: (self, encoded_tensors, decode_params, num_summands,
                               shape)
return_statement: return self.stage.decode(temp_encoded_tensors,
                               decode_params[EncoderKeys.PARAMS],
                               num_summands=num_summands,
                               shape=shape[EncoderKeys.SHAPE])
docstring: Implementation for the `decode_after_sum` method.
docstring_summary: Implementation for the `decode_after_sum` method.
docstring_tokens: Implementation for the decode_after_sum method .
function: def _decode_after_sum_impl(self, encoded_tensors, decode_params, num_summands,
                               shape):
      """Implementation for the `decode_after_sum` method."""
      if not self.stage.commutes_with_sum:
        # This should have been decoded earlier in the decode_before_sum method.
        assert tf.is_tensor(encoded_tensors)
        return encoded_tensors
  
      temp_encoded_tensors = {}
      for key, value in six.iteritems(encoded_tensors):
        if key in self.children:
          with tf.compat.v1.name_scope(None, '/'.join([self.stage.name, key])):
            temp_encoded_tensors[key] = self.children[key]._decode_after_sum_impl(  # pylint: disable=protected-access
                value, decode_params[EncoderKeys.CHILDREN][key], num_summands,
                shape[EncoderKeys.CHILDREN][key])
        else:
          temp_encoded_tensors[key] = value
      return self.stage.decode(temp_encoded_tensors,
                               decode_params[EncoderKeys.PARAMS],
                               num_summands=num_summands,
                               shape=shape[EncoderKeys.SHAPE])
function_tokens: def _decode_after_sum_impl ( self , encoded_tensors , decode_params , num_summands , shape ) : if not self . stage . commutes_with_sum : # This should have been decoded earlier in the decode_before_sum method. assert tf . is_tensor ( encoded_tensors ) return encoded_tensors temp_encoded_tensors = { } for key , value in six . iteritems ( encoded_tensors ) : if key in self . children : with tf . compat . v1 . name_scope ( None , '/' . join ( [ self . stage . name , key ] ) ) : temp_encoded_tensors [ key ] = self . children [ key ] . _decode_after_sum_impl ( # pylint: disable=protected-access value , decode_params [ EncoderKeys . CHILDREN ] [ key ] , num_summands , shape [ EncoderKeys . CHILDREN ] [ key ] ) else : temp_encoded_tensors [ key ] = value return self . stage . decode ( temp_encoded_tensors , decode_params [ EncoderKeys . PARAMS ] , num_summands = num_summands , shape = shape [ EncoderKeys . SHAPE ] )