UnsqueezeLayer.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 UnsqueezeLayer<T> : Layer<T>
    {
        public UnsqueezeLayer(CudaDnn<T> cuda, Log log, LayerParameter p)
            : base(cuda, log, p)
        {
            m_type = LayerParameter.LayerType.UNSQUEEZE;
        }
 
        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;
            List<int> rgDim = new List<int>(colBottom[0].shape());
 
            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);
            }
 
            rgIdx = rgIdx.OrderBy(p => p).ToList();
 
            for (int i = 0; i < rgIdx.Count; i++)
            {
                int nAxis = rgIdx[i];
 
                if (nAxis >= rgDim.Count)
                    rgDim.Add(1);
                else
                    rgDim.Insert(nAxis, 1);
            }
 
            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)
        {
        }
    }
}