tensorflow_proto

tensorflow_proto是一个使用prost-build生成Rust struct以用于Tensorflow使用协议缓冲区的序列化/反序列化操作的crate。

特别是，当在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"] }