tensorflow_proto

tensorflow_proto 是一个crate，使用 prost-build 生成Rust struct，用于Tensorflow在任何地方使用协议缓冲区的序列化/反序列化。

特别是，这在 tensorflow/rust 库中很有用，当调用 SessionOptions::set_config 来配置Tensorflow。

注意: 此crate针对tensorflow 1.15.2和2.0.0进行了测试。

用法

默认功能

添加

tensorflow_proto = "0.3.0"

到您的 Cargo.toml。

Serde支持

可以使用 "serde-derive" 功能启用Serde支持

tensorflow_proto = { version = "0.3.0", features = ["serde-derive"] }

这将向每个生成的 struct 添加一个 #[derive(serde::Serialize, serde::Deserialize)]。

您还必须依赖于 serde。

轻松转换为字节

最后，您可以为每个由prost生成的 struct 启用实现 std::convert::TryFrom 的代码生成，该实现将消息编码到 Vec<u8> 中。

tensorflow_proto = { version = "0.3.0", features = ["convert"] }

使用自定义Tensorflow *.proto 源

要使用Tensorflow协议缓冲区源的不同版本，定义 TENSORFLOW_PROTO_SOURCE 为Tensorflow源树的根。