예제 #1
0
        public static Tensor Tanh(Tensor input)
        {
            var output = new Tensor(input.Shape);

            GpuHelper.InvokeActivationKernel("Tanh", input, output);
            return(output);
        }
예제 #2
0
        public static Tensor LeakyRelu(Tensor input, float alpha)
        {
            var output = new Tensor(input.Shape);

            ComputeAssets.Activation.SetFloat("Alpha", alpha);
            GpuHelper.InvokeActivationKernel("LeakyRelu", input, output);
            return(output);
        }
예제 #3
0
파일: Math.cs 프로젝트: L4Z3RC47/Pix2Pix
 public static Tensor Tanh(Tensor input)
 {
     return(GpuHelper.InvokeActivationKernel("Tanh", input));
 }
예제 #4
0
파일: Math.cs 프로젝트: L4Z3RC47/Pix2Pix
 public static Tensor LeakyRelu(Tensor input, float alpha)
 {
     ComputeAssets.Activation.SetFloat("Alpha", alpha);
     return(GpuHelper.InvokeActivationKernel("LeakyRelu", input));
 }
예제 #5
0
파일: Math.cs 프로젝트: L4Z3RC47/Pix2Pix
 public static Tensor Relu(Tensor input)
 {
     return(GpuHelper.InvokeActivationKernel("Relu", input));
 }
예제 #6
0
 public static void Tanh(Tensor input, Tensor output)
 {
     GpuHelper.InvokeActivationKernel("Tanh", input, output);
 }