[−][src]Crate tract_tensorflow
Tract TensorFlow module
Tiny, no-nonsense, self contained, portable inference.
Example
use tract_core::prelude::*; // build a simple model that just add 3 to each input component let tf = tract_tensorflow::tensorflow(); let model = tf.model_for_path("tests/models/plus3.pb").unwrap(); // we build an execution plan. default input and output are inferred from // the model graph let plan = SimplePlan::new(&model).unwrap(); // run the computation. let input = tensor1(&[1.0f32, 2.5, 5.0]); let mut outputs = plan.run(tvec![input]).unwrap(); // take the first and only output tensor let mut tensor = outputs.pop().unwrap(); assert_eq!(tensor, rctensor1(&[4.0f32, 5.5, 8.0]));
Re-exports
pub use model::Tensorflow; |
Modules
model | |
ops | |
tensor | |
tfpb |
Functions
tensorflow |