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