ConstantLayer.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using MyCaffe.basecode;
using MyCaffe.common;
using MyCaffe.param;
 
namespace MyCaffe.layers
{
    public class ConstantLayer<T> : Layer<T>
    {
        BlobShape m_shape;
        List<float> m_rgF;
 
        public ConstantLayer(CudaDnn<T> cuda, Log log, LayerParameter p)
            : base(cuda, log, p)
        {
            m_type = LayerParameter.LayerType.CONSTANT;
        }
 
        public override void LayerSetUp(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
            m_shape = m_param.constant_param.output_shape;
            m_rgF = m_param.constant_param.values_f;
 
            if (!string.IsNullOrEmpty(m_param.constant_param.binary_data_file))
            {
                m_log.CHECK(File.Exists(m_param.constant_param.binary_data_file), "The 'binary_data_file' specified ('" + m_param.constant_param.binary_data_file + "') could not be found!");
 
                using (FileStream fs = new FileStream(m_param.constant_param.binary_data_file, FileMode.Open, FileAccess.Read))
                using (BinaryReader br = new BinaryReader(fs))
                {
                    BlobProto proto = BlobProto.Load(br);
                    m_rgF = proto.data;
                }
            }
        }
 
        public override void Reshape(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
            colTop[0].Reshape(m_param.constant_param.output_shape);
            T[] rgData = Utility.ConvertVec<T>(m_rgF.ToArray());
            colTop[0].SetData(rgData);
        }
 
        protected override void forward(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
        }
 
        protected override void backward(BlobCollection<T> colTop, List<bool> rgbPropagateDown, BlobCollection<T> colBottom)
        {            
        }
    }
}