Filler.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using MyCaffe.basecode;
using MyCaffe.param;
using MyCaffe.common;
 
namespace MyCaffe.fillers
{
    public abstract class Filler<T>
    {
        protected CudaDnn<T> m_cuda;
        protected Log m_log;
        protected FillerParameter m_param;
 
        public Filler(CudaDnn<T> cuda, Log log, FillerParameter p)
        {
            m_cuda = cuda;
            m_log = log;
            m_param = p;
        }
 
        public void Fill(Blob<T> b)
        {
            int nNum = (b.num_axes > 0) ? b.shape(0) : 1;
            int nNumChannels = (b.num_axes > 1) ? b.shape(1) : 1;
            int nHeight = (b.num_axes > 2) ? b.shape(2) : 1;
            int nWidth = (b.num_axes > 3) ? b.shape(3) : 1;
            Fill(b.count(), b.mutable_gpu_data, b.num_axes, nNum, nNumChannels, nHeight, nWidth);
        }
 
 
        public abstract void Fill(int nCount, long hMem, int nNumAxes = 1, int nNumOutputs = 1, int nNumChannels = 1, int nHeight = 1, int nWidth = 1);
 
        public static Filler<T> Create(CudaDnn<T> cuda, Log log, FillerParameter p)
        {
            switch (p.type)
            {
                case "constant":
                    return new ConstantFiller<T>(cuda, log, p);
 
                case "sequence":
                    return new SequenceFiller<T>(cuda, log, p);
 
                case "gaussian":
                    return new GaussianFiller<T>(cuda, log, p);
 
                case "uniform":
                    return new UniformFiller<T>(cuda, log, p);
 
                case "positive_unitball":
                    return new PositiveUnitballFiller<T>(cuda, log, p);
 
                case "xavier":
                    return new XavierFiller<T>(cuda, log, p);
 
                case "msra":
                    return new MsraFiller<T>(cuda, log, p);
 
                case "bilinear":
                    return new BilinearFiller<T>(cuda, log, p);
 
                default:
                    log.FAIL("Unknown filler type: " + p.type);
                    return null;
            }
        }
    }
}