Esempio n. 1
0
File: Math.cs Progetto: wangzy88/Ngx
        // ReLU activation function

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

        public static void Relu(Tensor input, Tensor output)
        {
            GpuBackend.InvokeActivation("Relu", input, 0, output);
        }
Esempio n. 3
0
File: Math.cs Progetto: 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);
        }
Esempio n. 4
0
        // Tanh activation function

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