public static TensorProto CreateTensor(float value)
        {
            var tensor = new TensorProto();

            tensor.FloatVal.Add(value);
            tensor.TensorShape = new TensorShapeProto();
            tensor.Dtype       = DataType.DtFloat;
            var dim = new TensorShapeProto.Types.Dim();

            dim.Size = 1;
            tensor.TensorShape.Dim.Add(dim);

            return(tensor);
        }
Exemplo n.º 2
0
        public static TensorProto CreateTensorFromString(string s)
        {
            s = s.Replace('/', '_').Replace('+', '-');

            var bytes = ByteString.CopyFrom(s, Encoding.ASCII);
            //Console.WriteLine(bytes.ToBase64());
            var tensor = new TensorProto();

            tensor.StringVal.Add(bytes);
            tensor.Dtype       = DataType.DtString;
            tensor.TensorShape = new TensorShapeProto();
            var dim = new TensorShapeProto.Types.Dim();

            dim.Size = 1;
            tensor.TensorShape.Dim.Add(dim);

            return(tensor);
        }