SqueezeParameter.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Linq;
using System.Text;
using MyCaffe.basecode;
 
namespace MyCaffe.param.beta
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class SqueezeParameter : LayerParameterBase
    {
        BlobShape m_shape = new BlobShape();
 
        public SqueezeParameter()
        {
            m_shape.dim = new List<int>();
        }
 
        [Description("Specifies the axes to remove if dim=1 on squeeze, or add dim=1 on unsqueeze.")]
        public List<int> axes
        {
            get { return m_shape.dim; }
            set { m_shape.dim = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            SqueezeParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            SqueezeParameter p = (SqueezeParameter)src;
            m_shape = p.m_shape.Clone();
        }
 
        public override LayerParameterBase Clone()
        {
            SqueezeParameter p = new SqueezeParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            return m_shape.ToProto(strName);
        }
 
        public static SqueezeParameter FromProto(RawProto rp)
        {
            SqueezeParameter p = new SqueezeParameter();
            p.m_shape = BlobShape.FromProto(rp);
 
            return p;
        }
    }
}