GramParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using MyCaffe.basecode;
using System.ComponentModel;
 
namespace MyCaffe.param.nt
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class GramParameter : LayerParameterBase 
    {
        int m_nAxis = 2;
        double m_dfAlpha = 1.0;
        double m_dfBeta = 1.0;
        bool m_bDisableScalingOnGradient = false;
 
        public GramParameter()
        {
        }
 
        [Description("The first axis to be lumped into a single Gram matrix computation -- may be negative to index from end (e.g., -1 for the last axis). For example, if axis == 2 and the input is (N x C x H x W), the output will be (N x C x C).")]
        public int axis
        {
            get { return m_nAxis; }
            set { m_nAxis = value; }
        }
 
        public double alpha
        {
            get { return m_dfAlpha; }
            set { m_dfAlpha = value; }
        }
 
        public double beta
        {
            get { return m_dfBeta; }
            set { m_dfBeta = value; }
        }
 
        public bool disable_scaling_on_gradient
        {
            get { return m_bDisableScalingOnGradient; }
            set { m_bDisableScalingOnGradient = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            GramParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            GramParameter p = (GramParameter)src;
            m_nAxis = p.m_nAxis;
            m_dfAlpha = p.m_dfAlpha;
            m_dfBeta = p.m_dfBeta;
            m_bDisableScalingOnGradient = p.m_bDisableScalingOnGradient;
        }
 
        public override LayerParameterBase Clone()
        {
            GramParameter p = new GramParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add("axis", axis.ToString());
            rgChildren.Add("alpha", alpha.ToString());
            rgChildren.Add("beta", beta.ToString());
            rgChildren.Add("disable_scaling_on_gradient", disable_scaling_on_gradient.ToString());
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static GramParameter FromProto(RawProto rp)
        {
            string strVal;
            GramParameter p = new GramParameter();
 
            if ((strVal = rp.FindValue("axis")) != null)
                p.axis = int.Parse(strVal);
 
            if ((strVal = rp.FindValue("alpha")) != null)
                p.alpha = ParseDouble(strVal);
 
            if ((strVal = rp.FindValue("beta")) != null)
                p.beta = ParseDouble(strVal);
 
            if ((strVal = rp.FindValue("disable_scaling_on_gradient")) != null)
                p.disable_scaling_on_gradient = bool.Parse(strVal);
 
            return p;
        }
    }
}