TileLayer.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 TileLayer<T> : Layer<T>
    {
        int m_nAxis;
        int m_nTiles;
        int m_nOuterDim;
        int m_nInnerDim;
 
        public TileLayer(CudaDnn<T> cuda, Log log, LayerParameter p)
            : base(cuda, log, p)
        {
            m_type = LayerParameter.LayerType.TILE;
        }
 
        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_nAxis = colBottom[0].CanonicalAxisIndex(m_param.tile_param.axis);
            m_nTiles = m_param.tile_param.tiles;
            m_log.CHECK_GT(m_param.tile_param.tiles, 0, "Number of tiles must be positive.");
 
            List<int> rgTopShape = Utility.Clone<int>(colBottom[0].shape());
            rgTopShape[m_nAxis] = colBottom[0].shape(m_nAxis) * m_nTiles;
            colTop[0].Reshape(rgTopShape);
 
            m_nOuterDim = colBottom[0].count(0, m_nAxis);
            m_nInnerDim = colBottom[0].count(m_nAxis);
        }
 
        protected override void forward(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
            long hBottomData = colBottom[0].gpu_data;
            long hTopData = colTop[0].mutable_gpu_data;
            int bottom_tile_axis = colBottom[0].shape(m_nAxis);
            int nCount = colTop[0].count();
 
            m_cuda.tile_fwd(nCount, hBottomData, m_nInnerDim, m_nTiles, bottom_tile_axis, hTopData);
        }
 
        protected override void backward(BlobCollection<T> colTop, List<bool> rgbPropagateDown, BlobCollection<T> colBottom)
        {
            if (!rgbPropagateDown[0])
                return;
 
            long hTopDiff = colTop[0].gpu_diff;
            long hBottomDiff = colBottom[0].mutable_gpu_diff;
            int bottom_tile_axis = colBottom[0].shape(m_nAxis);
            int tile_size = m_nInnerDim / bottom_tile_axis;
            int nCount = colBottom[0].count();
 
            m_cuda.tile_bwd(nCount, hTopDiff, tile_size, m_nTiles, bottom_tile_axis, hBottomDiff);
        }
    }
}