Struct autograd::Eval

pub struct Eval<'view, 'feed, 'graph, F: Float> { /* fields omitted */ }

Helper structure for batched evaluation.

Eval structure can buffer evaluation targets with useful push and extend functions and runs batched evaluation. Use this in case Tensor::eval or Graph::eval doesn’t help.

use autograd as ag;
use ndarray;

ag::with(|g| {
   let a = g.placeholder(&[]);
   let x = a + a;
   let y = a * a;
   let z = a / a;

   ag::Eval::new(g)
       .push(&x)
       .extend(&[y, z])
       .feed(&[a.given(ndarray::arr0(2.).view())])
       .run();  // Do eval
   });

Implementations

impl<'feed, 'tensor, 'view, 'graph, F: Float> Eval<'view, 'feed, 'graph, F>

pub fn new(scope: &'graph Graph<F>) -> Self

Instantiates a new evaluation session.

pub fn push<A>(&mut self, x: A) -> &mut Self where
    A: AsRef<Tensor<'graph, F>>, 

Appends a tensor to the back of the evaluation targets.

pub fn feed(&mut self, feeds: &'feed [Feed<'view, F>]) -> &mut Self

feeds is a sequence of (placeholder-tensor, its value)

pub fn extend<A>(&mut self, xs: &'tensor [A]) -> &mut Self where
    A: AsRef<Tensor<'graph, F>>, 

Extends the evaluation targets with xs.

pub fn run(&'tensor self) -> Vec<Result<NdArray<F>, EvalError>>

Evaluates the buffered tensors.