PositiveUnitballFiller.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 PositiveUnitballFiller<T> : Filler<T>
    {
        public PositiveUnitballFiller(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_cuda.rng_uniform(nCount, 0, 1, hMem);
 
            // We expect the filler to not be called very frequently, so we will
            // just use a simple implementation.
 
            T[] rgData = m_cuda.GetMemory(hMem);
            int nDim = nCount / nNumOutputs;
 
            m_log.CHECK_GT(nDim, 0, "The dimension must be greater than 0.");
 
            for (int i = 0; i < nNumOutputs; i++)
            {
                double dfSum = 0;
 
                for (int j = 0; j < nDim; j++)
                {
                    int nIdx = i * nDim + j;
                    dfSum += (double)Convert.ChangeType(rgData[nIdx], typeof(double));
                }
 
                for (int j = 0; j < nDim; j++)
                {
                    int nIdx = i * nDim + j;
                    double dfVal = (double)Convert.ChangeType(rgData[nIdx], typeof(double)) / dfSum;
                    rgData[nIdx] = (T)Convert.ChangeType(dfVal, typeof(T));
                }
            }
 
            m_cuda.SetMemory(hMem, rgData);
 
            m_log.CHECK_EQ(-1, m_param.sparse, "Sparsity not supported by this Filler.");
        }
    }
}