Function dfdx::tensor_ops::ne

source ·
pub fn ne<S: Shape, E, D: CmpKernel<NeKernelOp, 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 inequality 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.ne(&b);
assert_eq!(r.array(), [false, true, true, false]);
let a = dev.tensor([-2, -1, 0, 1, 2]);
let r = a.ne(1);
assert_eq!(r.array(), [true, true, true, false, true]);