FlattenLayer.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using MyCaffe.basecode;
using MyCaffe.common;
using MyCaffe.param;
 
namespace MyCaffe.layers
{
    public class FlattenLayer<T> : Layer<T>
    {
        public FlattenLayer(CudaDnn<T> cuda, Log log, LayerParameter p)
            : base(cuda, log, p)
        {
            m_type = LayerParameter.LayerType.FLATTEN;
        }
 
        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)
        {
            m_log.CHECK(colTop[0] != colBottom[0], "Layer does not allow in-place computation.");
 
            int nStartAxis = colBottom[0].CanonicalAxisIndex(m_param.flatten_param.axis);
            int nEndAxis = colBottom[0].CanonicalAxisIndex(m_param.flatten_param.end_axis);
 
            //List<int> rgTopShape = new List<int>();
            //for (int i = 0; i < nStartAxis; i++)
            //{
            //    rgTopShape.Add(colBottom[0].shape(i));
            //}
 
            //int nFlattenDim = colBottom[0].count(nStartAxis, nEndAxis + 1);
            //rgTopShape.Add(nFlattenDim);
 
            //for (int i = nEndAxis + 1; i < colBottom[0].num_axes; i++)
            //{
            //    rgTopShape.Add(colBottom[0].shape(i));
            //}
 
            List<int> rgTopShape = FlattenParameter.Reshape(m_param.flatten_param.axis, m_param.flatten_param.end_axis, colBottom[0].shape(), nStartAxis, nEndAxis);
 
            colTop[0].Reshape(rgTopShape);
            m_log.CHECK_EQ(colTop[0].count(), colBottom[0].count(), "The top[0] and bottom[0] should have the same count.");
        }
 
        protected override void forward(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
            m_cuda.copy(colTop[0].count(), colBottom[0].gpu_data, colTop[0].mutable_gpu_data);
//            colTop[0].ShareData(colBottom[0]);
        }
 
        protected override void backward(BlobCollection<T> colTop, List<bool> rgbPropagateDown, BlobCollection<T> colBottom)
        {
            m_cuda.copy(colBottom[0].count(), colTop[0].gpu_diff, colBottom[0].mutable_gpu_diff);
//            colBottom[0].ShareDiff(colTop[0]);
        }
    }
}