Function apply_elementwise_operation 

pub fn apply_elementwise_operation<F>(
    input: &Tensor,
    _inplace: bool,
    operation: F,
    _context: &str,
) -> Result<Tensor>
where
    F: Fn(f32) -> f32,

Utility for element-wise operations with inplace support