UniformFiller.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 class UniformFiller<T> : Filler<T>
    {
        public UniformFiller(CudaDnn<T> cuda, Log log, FillerParameter p)
            : base(cuda, log, p)
        {
        }
 
        public override void Fill(int nCount, long hMem, int nNumAxes = 1, int nNumOutputs = 1, int nNumChannels = 1, int nHeight = 1, int nWidth = 1)
        {
            m_log.CHECK(nCount > 0, "There is no data to fill!");
            m_log.CHECK_LT(m_param.min, m_param.max, "The min must be less than the max for the uniform filler!");
            m_cuda.rng_uniform(nCount, m_param.min, m_param.max, hMem);
        }
    }
}