def add_axis_mean(self, input_name, axis, output_name):
layer = caffe_net.LayerParameter(name=output_name,
type='AxisMean',
bottom=[input_name],
top=[output_name])
layer.axis_mean_param(axis)
self.caffe_model.add_layer(layer)
self.data_dict[output_name] = Operators.axis_mean(
self.data_dict[input_name], axis, output_name)
return output_name
