extern crate tensorflux;

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

macro_rules! ok(($result:expr) => ($result.unwrap()));

fn main() {
    let graph = "examples/assets/multiplication.pb"; // c = a * b
    let mut session = ok!(Session::new(&ok!(Options::new())));
    ok!(session.extend(&ok!(Buffer::load(graph))));

    let a = ok!(Tensor::new(vec![1f32, 2.0, 3.0], &[3]));
    let b = ok!(Tensor::new(vec![4f32, 5.0, 6.0], &[3]));

    let inputs = vec![Input::new("a", a), Input::new("b", b)];
    let mut outputs = vec![Output::new("c")];
    ok!(session.run(&inputs, &mut outputs, &[], None, None));

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