Function compare 

pub fn compare(lhs: &Tensor, rhs: &Tensor, dir: &CompareDir) -> Result<Tensor>

Compare two CPU tensors elementwise.

Examples

use tenferro_cpu::compare;
use tenferro_tensor::{CompareDir, 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 = compare(&a, &b, &CompareDir::Gt)?;
assert_eq!(out.as_slice::<bool>().unwrap(), &[false, true]);