Function autograd::ops::not_equal
[−]
[src]
pub fn not_equal(a: &Tensor, b: &Tensor) -> Tensor
Compares two tensors and returns a binary tensor.
if a[i] != b[i]
then return_value[i]
will be 1 else 0
Panics
When broadcast is impossible
extern crate ndarray; extern crate autograd as ag; let mut ctx = ag::Context::new(); let ref a = ag::constant(ndarray::arr1(&[1., 2., 3.]), &mut ctx); let ref b = ag::constant(ndarray::arr1(&[3., 2., 1.]), &mut ctx); let ref c = ag::not_equal(a, b); assert_eq!(c.eval(&mut ctx), ndarray::arr1(&[1., 0., 1.]).into_dyn());