SqueezeLayer.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using MyCaffe.basecode;
using MyCaffe.common;
using MyCaffe.param;
using MyCaffe.param.beta;
 
namespace MyCaffe.layers.beta
{
    public class SqueezeLayer<T> : Layer<T>
    {
        public SqueezeLayer(CudaDnn<T> cuda, Log log, LayerParameter p)
            : base(cuda, log, p)
        {
            m_type = LayerParameter.LayerType.SQUEEZE;
        }
 
        public override int ExactNumBottomBlobs
        {
            get { return 1; }
        }
 
        public override int ExactNumTopBlobs
        {
            get { return 1; }
        }
 
        public override void LayerSetUp(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
        }
 
        public override void Reshape(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
            if (!reshapeNeeded(colBottom, colTop))
                return;
 
            float[] rgIdxF = null;
 
            if (colBottom.Count > 1)
                rgIdxF = convertF(colBottom[1].mutable_cpu_data);
 
            List<int> rgIdx = new List<int>();
            if (rgIdxF != null)
            {
                if (m_param.squeeze_param.axes.Count > 0)
                    m_log.WriteLine("WARNING: Squeeze indexes specified in as blob input data, squeeze parameters will be ignored.");
 
                for (int i = 0; i < rgIdxF.Length; i++)
                {
                    rgIdx.Add((int)rgIdxF[i]);
                }
            }
            else
            {
                rgIdx = new List<int>(m_param.squeeze_param.axes);
            }
 
            List<int> rgDim = new List<int>();
            int nIdx = 0;
 
            for (int i = 0; i < colBottom[0].shape().Count; i++)
            {
                int nDim = colBottom[0].shape()[i];
 
                if (nDim != 1)
                {
                    rgDim.Add(nDim);
                }
                else if (rgIdx != null && rgIdx.Count > 0)
                {
                    if (nIdx >= rgIdx.Count)
                    {
                        rgDim.Add(nDim);
                    }
                    else
                    {
                        if (rgIdx[nIdx] != i)
                            rgDim.Add(nDim);
                        nIdx++;
                    }
                }
            }
 
            colTop[0].Reshape(rgDim);
            colTop[0].ShareData(colBottom[0]);
            colTop[0].ShareDiff(colBottom[0]);
        }
 
        protected override void forward(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
        }
 
        protected override void backward(BlobCollection<T> colTop, List<bool> rgbPropagateDown, BlobCollection<T> colBottom)
        {
        }
    }
}