tensorflux 0.2.1

The package provides an interface to TensorFlow.
docs.rs failed to build tensorflux-0.2.1
Please check the build logs for more information.
See Builds for ideas on how to fix a failed build, or Metadata for how to configure docs.rs builds.
If you believe this is docs.rs' fault, open an issue.

TensorFlux Version Status

The package provides an interface to TensorFlow.

Documentation

Example

Create a graph in Python:

import tensorflow as tf

with tf.Session() as session:
    a = tf.Variable(0.0, name='a')
    b = tf.Variable(0.0, name='b')
    c = tf.mul(a, b, name='c')
    tf.train.write_graph(session.graph_def, '', 'graph.pb', as_text=False)

Evaluate the graph in Rust:

use tensorflux::{Definition, Input, Options, Output, Session, Tensor};

let mut session = Session::new(&Options::new().unwrap()).unwrap();
session.extend(&Definition::load(GRAPH_PATH).unwrap()).unwrap(); // c = a * b

let mut inputs = vec![Input::new("a"), Input::new("b")];
inputs[0].set(Tensor::new(vec![1f32, 2.0, 3.0], &[3]).unwrap());
inputs[1].set(Tensor::new(vec![4f32, 5.0, 6.0], &[3]).unwrap());

let mut outputs = vec![Output::new("c")];

session.run(&mut inputs, &mut outputs, &vec![]).unwrap();

let result = outputs[0].get::<f32>().unwrap();
assert_eq!(&result[..], &[1.0 * 4.0, 2.0 * 5.0, 3.0 * 6.0]);

Requirements

Configuration

Contribution

Your contribution is highly appreciated. Do not hesitate to open an issue or a pull request. Note that any contribution submitted for inclusion in the project will be licensed according to the terms given in LICENSE.md.