ClipLayer.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 ClipLayer<T> : NeuronLayer<T>
    {
        public ClipLayer(CudaDnn<T> cuda, Log log, LayerParameter p)
            : base(cuda, log, p)
        {
            m_type = LayerParameter.LayerType.CLIP;
        }
 
        protected override void forward(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
            long hBottomData = colBottom[0].gpu_data;
            long hTopData = colTop[0].mutable_gpu_data;
            int nCount = colBottom[0].count();
            double dfMin = m_param.clip_param.min;
            double dfMax = m_param.clip_param.max;
 
            m_cuda.clip_fwd(nCount, hBottomData, hTopData, (T)Convert.ChangeType(dfMin, typeof(T)), (T)Convert.ChangeType(dfMax, typeof(T)));
        }
 
        protected override void backward(BlobCollection<T> colTop, List<bool> rgbPropagateDown, BlobCollection<T> colBottom)
        {
            if (rgbPropagateDown[0])
            {
                long hBottomData = colBottom[0].gpu_data;
                long hTopDiff = colTop[0].gpu_diff;
                long hBottomDiff = colBottom[0].mutable_gpu_diff;
                int nCount = colBottom[0].count();
                double dfMin = m_param.clip_param.min;
                double dfMax = m_param.clip_param.max;
 
                m_cuda.clip_bwd(nCount, hTopDiff, hBottomData, hBottomDiff, (T)Convert.ChangeType(dfMin, typeof(T)), (T)Convert.ChangeType(dfMax, typeof(T)));
            }
        }
    }
}