Function select 

pub fn select(
    pred: &Tensor,
    on_true: &Tensor,
    on_false: &Tensor,
) -> Result<Tensor>

Select values from two tensors using a boolean predicate tensor.

Examples

use tenferro_cpu::select;
use tenferro_tensor::Tensor;

let pred = Tensor::from_vec_col_major(vec![2], vec![true, false])?;
let on_true = Tensor::from_vec_col_major(vec![2], vec![1.0_f64, 2.0])?;
let on_false = Tensor::from_vec_col_major(vec![2], vec![3.0_f64, 4.0])?;
let out = select(&pred, &on_true, &on_false)?;
assert_eq!(out.as_slice::<f64>().unwrap(), &[1.0, 4.0]);