C# (CSharp) GpuBackend.InvokeBatchNorm Examples

Example #1

File: Math.cs Project: 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);
        }

Example #2

File: Math.cs Project: 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);
        }