Exemplo n.º 1
0
        public static Tensor conv2d(object parameters)
        {
            var args = Python.ConvertToDict(parameters);

            var input            = args["input"];
            var filter           = args["filter"];
            var strides          = args["strides"];
            var padding          = args["padding"];
            var name             = args["name"];
            var data_format      = args.ContainsKey("data_format") ? args["data_format"] : "NHWC";
            var use_cudnn_on_gpu = args.ContainsKey("use_cudnn_on_gpu") ? args["use_cudnn_on_gpu"] : true;
            var dilations        = args.ContainsKey("dilations") ? args["dilations"] : new int[] { 1, 1, 1, 1 };

            var _op = _op_def_lib._apply_op_helper("Conv2D", name: name?.ToString(), args: new
            {
                input,
                filter,
                strides,
                padding,
                use_cudnn_on_gpu,
                data_format,
                dilations
            });

            return(_op.outputs[0]);
        }
Exemplo n.º 2
0
 public static Dictionary <string, object> ConvertToDict(object dyn)
 => Python.ConvertToDict(dyn);