ReductionParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.ComponentModel;
using MyCaffe.basecode;
 
namespace MyCaffe.param
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class ReductionParameter : LayerParameterBase
    {
        public enum ReductionOp
        {
            SUM = 1,
            ASUM = 2,
            SUMSQ = 3,
            MEAN = 4,
            MIN = 5,
            MAX = 6
        }
 
        ReductionOp m_operation = ReductionOp.SUM;
        int m_nAxis = 0;
        double m_dfCoeff = 1.0;
 
        public ReductionParameter()
        {
        }
 
        [Description("Specifies the reduction operation.")]
        public ReductionOp operation
        {
            get { return m_operation; }
            set { m_operation = value; }
        }
 
        [Description("Specifies the first axis to reduce to scalar -- may be negative to index from the end (e.g., -1 for the last axis)..")]
        public int axis
        {
            get { return m_nAxis; }
            set { m_nAxis = value; }
        }
 
        [Description("Specifies the coefficient for output.")]
        public double coeff
        {
            get { return m_dfCoeff; }
            set { m_dfCoeff = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            ReductionParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            ReductionParameter p = (ReductionParameter)src;
            m_operation = p.m_operation;
            m_nAxis = p.m_nAxis;
            m_dfCoeff = p.m_dfCoeff;
        }
 
        public override LayerParameterBase Clone()
        {
            ReductionParameter p = new ReductionParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add("operation", operation.ToString());
 
            if (axis != 0)
                rgChildren.Add("axis", axis.ToString());
 
            if (coeff != 1.0)
                rgChildren.Add("coeff", coeff.ToString());
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static ReductionParameter FromProto(RawProto rp)
        {
            string strVal;
            ReductionParameter p = new ReductionParameter();
 
            if ((strVal = rp.FindValue("operation")) != null)
            {
                switch (strVal)
                {
                    case "ASUM":
                        p.operation = ReductionOp.ASUM;
                        break;
 
                    case "MEAN":
                        p.operation = ReductionOp.MEAN;
                        break;
 
                    case "SUM":
                        p.operation = ReductionOp.SUM;
                        break;
 
                    case "SUMSQ":
                        p.operation = ReductionOp.SUMSQ;
                        break;
 
                    case "MIN":
                        p.operation = ReductionOp.MIN;
                        break;
 
                    case "MAX":
                        p.operation = ReductionOp.MAX;
                        break;
 
                    default:
                        throw new Exception("Uknown 'operation' value: " + strVal);
                }
            }
 
            if ((strVal = rp.FindValue("axis")) != null)
                p.axis = int.Parse(strVal);
 
            if ((strVal = rp.FindValue("coeff")) != null)
                p.coeff = ParseDouble(strVal);
 
            return p;
        }
    }
}