Exemplos de GpuBackend.InvokeBatchNorm em C# (CSharp)

Exemplo n.º 1

Arquivo: Math.cs Projeto: wangzy88/Ngx

        public static Tensor BatchNorm(Tensor input, Tensor scale, Tensor offset)
        {
            var output = new Tensor(input.Shape);

            GpuBackend.InvokeBatchNorm(input, scale, offset, output);
            return(output);
        }

Exemplo n.º 2

Arquivo: Math.cs Projeto: wangzy88/Ngx

        // Batch normalization

        public static void BatchNorm(Tensor input, Tensor scale, Tensor offset, Tensor output)
        {
            output.Reset(input.Shape);
            GpuBackend.InvokeBatchNorm(input, scale, offset, output);
        }