Function autograd::ops::equal

pub fn 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::equal(a, b);

assert_eq!(c.eval(&mut ctx), ndarray::arr1(&[0., 1., 0.]).into_dyn());