Function dfdx::tensor_ops::maximum

source ·

pub fn maximum<S: Shape, E: Dtype, D: Device<E>, LTape: Tape<E, D> + Merge<R>, R: Default>(
    lhs: Tensor<S, E, D, LTape>,
    rhs: Tensor<S, E, D, R>
) -> Tensor<S, E, D, LTape>

Expand description

Element wise maximum.

Pytorch equivalent: torch.maximum(a, b)

Example:

let a = dev.tensor([[1.0, 2.0, 3.0], [-1.0, -2.0, -3.0]]);
let b = dev.tensor([[1.0, 0.5, 1.0], [-2.0, 2.0, -3.5]]);
let r = a.maximum(b);
assert_eq!(r.array(), [[1.0, 2.0, 3.0], [-1.0, 2.0, -3.0]]);