예제 #1
0
        public static Tensor <float> Create(
            Tensor <float> x,
            Tensor <float> y,
            Func <Tensor <float>, Tensor <float>, Tensor <float> > f,
            Dim convDim,
            string description
            )
        {
            var _x = new Var(x.Shape.DropLeft(1), "_x");
            var _y = new Var(y.Shape.DropLeft(1), "_y");
            var _z = f(_x, _y);

            var _shape = convDim.Pad(_z.Shape);
            var lambda = new Lambda {
                Expr = _z, Vars = new[] { _x, _y }
            };

            return(new CorrelateCustom(x, y, _x, _y, _z, f, _shape, lambda, description));
        }