pub fn clamp(input: &Tensor, lower: &Tensor, upper: &Tensor) -> Result<Tensor>Expand description
Clamp CPU tensor values elementwise between lower and upper bounds.
ยงExamples
use tenferro_cpu::clamp;
use tenferro_tensor::Tensor;
let input = Tensor::from_vec_col_major(vec![3], vec![-1.0_f64, 2.0, 8.0])?;
let lower = Tensor::from_vec_col_major(vec![3], vec![0.0_f64, 0.0, 0.0])?;
let upper = Tensor::from_vec_col_major(vec![3], vec![5.0_f64, 5.0, 5.0])?;
let out = clamp(&input, &lower, &upper)?;
assert_eq!(out.as_slice::<f64>().unwrap(), &[0.0, 2.0, 5.0]);