Skip to main content

maximum

Function maximum 

Source
pub fn maximum(lhs: &Tensor, rhs: &Tensor) -> Result<Tensor>
Expand description

Compute elementwise maximum values.

ยงExamples

use tenferro_cpu::maximum;
use tenferro_tensor::Tensor;

let a = Tensor::from_vec_col_major(vec![2], vec![1.0_f64, 5.0])?;
let b = Tensor::from_vec_col_major(vec![2], vec![3.0_f64, 4.0])?;
let out = maximum(&a, &b)?;
assert_eq!(out.as_slice::<f64>().unwrap(), &[3.0, 5.0]);