예제 #1
0
파일: Math.cs 프로젝트: wangzy88/Ngx
        // ReLU activation function

        public static void Relu(Tensor input, Tensor output)
        {
            output.Reset(input.Shape);
            GpuBackend.InvokeActivation("Relu", input, 0, output);
        }
예제 #2
0
        // ReLU activation function

        public static void Relu(Tensor input, Tensor output)
        {
            GpuBackend.InvokeActivation("Relu", input, 0, output);
        }
예제 #3
0
파일: Math.cs 프로젝트: wangzy88/Ngx
        // Leaky ReLU activation function

        public static void LeakyRelu(Tensor input, float alpha, Tensor output)
        {
            output.Reset(input.Shape);
            GpuBackend.InvokeActivation("LeakyRelu", input, alpha, output);
        }
예제 #4
0
        // Tanh activation function

        public static void Tanh(Tensor input, Tensor output)
        {
            GpuBackend.InvokeActivation("Tanh", input, 0, output);
        }