Add try_unary_mut

[viirya]

Is your feature request related to a problem or challenge? Please describe what you are trying to do.

We've added the support of mutating primitive array. Now we can try to use it on compute kernels.
This adds try_unary_mut so users can implement these kernels by themselves. This can benefit our use cases by avoid extra memory allocation when applying kernels on arrays.

Describe the solution you'd like

Describe alternatives you've considered

Additional context