Function dfdx::tensor_ops::eq

pub fn eq<S: Shape, E, D: CmpKernel<EqKernelOp, E>, T: Tape<E, D>>(
    lhs: &Tensor<S, E, D, T>,
    rhs: &Tensor<S, E, D, T>
) -> Tensor<S, bool, D, NoneTape>

Element-wise equality comparison. ==

Examples:

let a = dev.tensor([1.2, 3.4, -0.5, -6.7]);
let b = dev.tensor([1.2, 0.0, 3.14, -6.7]);
let r = a.eq(&b);
assert_eq!(r.array(), [true, false, false, true]);

let a = dev.tensor([-2, -1, 0, 1, 2]);
let r = a.eq(1);
assert_eq!(r.array(), [false, false, false, true, false]);