C# (CSharp) Pix2Pix GpuHelper.InvokeConcatKernel 예제들

예제 #1

파일: Math.cs 프로젝트: L4Z3RC47/Pix2Pix

        public static Tensor Concat(Tensor input1, Tensor input2)
        {
            var elements = input1.Shape[0] * input1.Shape[1];
            var kernel   = elements < 512 ? "Concat64" : "Concat512";

            if (elements < 64)
            {
                kernel = "Concat4";
            }
            return(GpuHelper.InvokeConcatKernel(kernel, input1, input2));
        }

예제 #2

 public static Tensor Concat(Tensor input1, Tensor input2)
 {
     return(GpuHelper.InvokeConcatKernel(input1, input2));
 }