Function optimized_sub 

pub fn optimized_sub<T>(a: &Tensor<T>, b: &Tensor<T>) -> Result<Tensor<T>>
where
    T: Clone + Default + Zero + StdSub<Output = T> + Send + Sync + 'static + Pod,