pub fn sub(lhs: &Tensor, rhs: &Tensor) -> Result<Tensor>Expand description
Subtract two CPU tensors elementwise.
ยงExamples
use tenferro_cpu::sub;
use tenferro_tensor::Tensor;
let a = Tensor::from_vec_col_major(vec![2], vec![5.0_f64, 2.0])?;
let b = Tensor::from_vec_col_major(vec![2], vec![3.0_f64, 4.0])?;
let out = sub(&a, &b)?;
assert_eq!(out.as_slice::<f64>().unwrap(), &[2.0, -2.0]);