Skip to main content

add

Function add 

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

Add two CPU tensors elementwise.

ยงExamples

use tenferro_cpu::add;
use tenferro_tensor::Tensor;

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