Function dfdx::tensor_ops::maximum
source · [−]pub fn maximum<Lhs, Rhs>(lhs: Lhs, rhs: Rhs) -> Lhswhere
Lhs: Tensor<Dtype = f32>,
Rhs: Tensor<Dtype = f32, Array = Lhs::Array>,
Lhs::Tape: Merge<Rhs::Tape>,
Expand description
Element wise maximum.
Pytorch equivalent: torch.maximum(a, b)
Example:
let a = tensor([[1.0, 2.0, 3.0], [-1.0, -2.0, -3.0]]);
let b = tensor([[1.0, 0.5, 1.0], [-2.0, 2.0, -3.5]]);
let r = a.maximum(b);
assert_eq!(r.data(), &[[1.0, 2.0, 3.0], [-1.0, 2.0, -3.0]]);