Yup, as long as it is valid mathematical operations you can process any vector, matrices, tensors depend on your need. The only problem usually is how to write it in valid python code, so we need more example :)
Discussion
on python i would use first numba with a custom kernel to try it out, improve it... after that switch to C