Function unary_expand

pub fn unary_expand<F>(
    context: &mut CubeContext,
    input: ExpandElement,
    func: F,
) -> ExpandElement
where
    F: Fn(UnaryOperator) -> Operator,