code: def softmax(attrs, inputs, proto_obj):
      
      if 'axis' not in attrs:
          attrs = translation_utils._add_extra_attributes(attrs, {'axis': 1})
      return 'softmax', attrs, inputs
docs: Softmax function.