Function autograd::ops::equal
[−]
[src]
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());