Function autograd::ops::setdiff1d

pub fn setdiff1d(a: &Tensor, b: &Tensor) -> Tensor

Takes diff between two tensors

Returns the sorted, unique values in a that are not in b.

extern crate ndarray;
extern crate autograd as ag;

let mut ctx = ag::Context::new();
let ref a = ag::constant(ndarray::arr1(&[4., 1., 5., 2., 3., 6.]), &mut ctx);
let ref b = ag::constant(ndarray::arr2(&[[2., 3.], [1., 4.]]), &mut ctx);
let ref c = ag::setdiff1d(a, b);

assert_eq!(c.eval(&mut ctx).as_slice().unwrap(), &[5., 6.])