Function dfdx::tensor_ops::lt

source ·
pub fn lt<S: Shape, E, D: CmpKernel<LtKernelOp, 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 strictly less than comparison. <

Examples:

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