Function autograd::ops::convert_to_tensor [] [src]

pub fn convert_to_tensor<T>(arr: Array<f32, T>) -> Tensor where
    T: Dimension

Converts ndarray::Array to ag::Tensor.

extern crate ndarray;
extern crate autograd as ag;

let arr = ndarray::arr1(&[2., 3.]);
let tensor = ag::convert_to_tensor(arr.clone());
assert_eq!(tensor.eval(&mut ag::Context::new()), arr.into_dyn());