CopyLayer.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 CopyLayer<T> : Layer<T>
    {
        public CopyLayer(CudaDnn<T> cuda, Log log, LayerParameter p)
            : base(cuda, log, p)
        {
            m_type = LayerParameter.LayerType.COPY;
        }
 
        public override int ExactNumBottomBlobs 
        {
            get { return 2; }
        }
 
        public override int ExactNumTopBlobs
        {
            get { return 0; }
        }
 
        public override void LayerSetUp(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
        }
 
        public override void Reshape(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
        }
 
        protected override void forward(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
            m_log.CHECK_EQ(colBottom[0].count(), colBottom[1].count(), "The bottom(0) and bottom(1) must have the same count!");
            m_cuda.copy(colBottom[0].count(), colBottom[0].gpu_data, colBottom[1].mutable_gpu_data);
        }
 
        protected override void backward(BlobCollection<T> colTop, List<bool> rgbPropagateDown, BlobCollection<T> colBottom)
        {
            m_log.CHECK_EQ(colBottom[0].count(), colBottom[1].count(), "The bottom(0) and bottom(1) must have the same count!");
            m_cuda.copy(colBottom[1].count(), colBottom[1].gpu_diff, colBottom[0].mutable_gpu_diff);
        }
    }
}