Function dfdx::tensor_ops::ge

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