ParameterLayer.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 ParameterLayer<T> : NeuronLayer<T>
    {
        public ParameterLayer(CudaDnn<T> cuda, Log log, LayerParameter p)
            : base(cuda, log, p)
        {
            m_type = LayerParameter.LayerType.PARAMETER;
        }
 
        public override int ExactNumBottomBlobs
        {
            get { return 0; }
        }
 
        public override int ExactNumTopBlobs
        {
            get { return 1; }
        }
 
        public override void LayerSetUp(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
            if (blobs.Count > 0)
            {
                m_log.WriteLine("Skipping parameter initialization.");
            }
            else
            {
                Blob<T> blob = new Blob<T>(m_cuda, m_log);
                blob.Name = m_param.name;
                blob.Reshape(m_param.parameter_param.shape);
                m_colBlobs.Add(blob);
            }
 
            m_bUseHalfSize = m_param.use_halfsize;
 
            colTop[0].Reshape(m_param.parameter_param.shape, m_bUseHalfSize);
        }
 
        public override void Reshape(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
        }
 
        protected override void forward(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
            colTop[0].ShareData(m_colBlobs[0]);
            colTop[0].ShareDiff(m_colBlobs[0]);
        }
 
        protected override void backward(BlobCollection<T> colTop, List<bool> rgbPropagateDown, BlobCollection<T> colBottom)
        {
        }
    }
}