GatherParameter.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 GatherParameter : LayerParameterBase
    {
        int m_nAxis = 0;
 
        public GatherParameter()
        {
        }
 
        [Description("Specifies the first axis to gather: all preceding axes are retained in the output.")]
        public int axis
        {
            get { return m_nAxis; }
            set { m_nAxis = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            GatherParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            GatherParameter p = (GatherParameter)src;
            m_nAxis = p.m_nAxis;
        }
 
        public override LayerParameterBase Clone()
        {
            GatherParameter p = new GatherParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add("axis", axis.ToString());
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static GatherParameter FromProto(RawProto rp)
        {
            string strVal;
            GatherParameter p = new GatherParameter();
 
            if ((strVal = rp.FindValue("axis")) != null)
                p.axis = int.Parse(strVal);
 
            return p;
        }
 
        public static List<int> Reshape(int nAxis, List<int> rgBtmShape, List<int> rgIdxShape, float[] rgIdxF, out int nDim, out int nDimAtAxis, out int nM, out int nN, out string strErr)
        {
            nAxis = Utility.CanonicalAxisIndex(nAxis, rgBtmShape.Count);
            nDim = Utility.Count(rgBtmShape, nAxis + 1);
            nDimAtAxis = rgBtmShape[nAxis];
            nM = Utility.Count(rgBtmShape, 0, nAxis);
            nN = Utility.Count(rgIdxShape);
 
            strErr = null;
 
            if (rgIdxF != null)
            {
                if (nN != rgIdxF.Length)
                {
                    strErr = "N should equal the number of indices.";
                    return null;
                }
 
                for (int i = 0; i < nN; i++)
                {
                    int nIdx = (int)rgIdxF[i];
                    if (nIdx < -nDimAtAxis || nIdx > nDimAtAxis)
                    {
                        strErr = "The index at idx=" + i.ToString() + " is out of range!  Must be within range [-" + nDimAtAxis.ToString() + "," + nDimAtAxis.ToString() + "]";
                        return null;
                    }
                }
            }
 
            List<int> rgTopShape = new List<int>(rgIdxShape);
            int nLen = rgTopShape.Count;
 
            while (rgTopShape.Count > 0 && rgTopShape[rgTopShape.Count - 1] == 1)
            {
                rgTopShape.RemoveAt(rgTopShape.Count - 1);
            }
 
            if (nAxis == 0)
                rgTopShape.Add(nDim);
            else if (nAxis == 1)
                rgTopShape.Insert(0, nM);
 
            for (int i = rgTopShape.Count; i < nLen; i++)
            {
                rgTopShape.Add(1);
            }
 
            return rgTopShape;
        }
    }
}