Function where_cond

pub fn where_cond(x: &Tensor, input: &Tensor, other: &Tensor) -> Tensor