Skip to main content

mul

Function mul 

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

Multiply two CPU tensors elementwise.

ยงExamples

use tenferro_cpu::mul;
use tenferro_tensor::Tensor;

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