Exemplo n.º 1
0
        public IWeightMatrix Process(IWeightMatrix input, IComputeGraph innerGraph)
        {
            var alphas = innerGraph.RepeatRows(alpha, input.Rows);
            var betas  = innerGraph.RepeatRows(beta, input.Rows);

            return(innerGraph.LayerNorm(input, alphas, betas));
        }
Exemplo n.º 2
0
 public IWeightTensor Norm(IWeightTensor input, IComputeGraph g)
 {
     return(g.LayerNorm(input, m_alpha, m_beta));
 }
Exemplo n.º 3
0
 public IWeightTensor Norm(IWeightTensor input, IComputeGraph g)
 {
     return(g.LayerNorm(input, this.m_alpha, this.m_beta, 1e-06f));
 }