Function select 

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