Function dfdx::tensor_ops::eq
source · 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>
Expand description
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]);