Does the controller have a version that is compatible with Tensor?

We are using the omnigymenv framework(gpu) and the get_observation() data obtained is tensor. What I understand is that the controller is not compatible with tensor but with np. We need to convert tensor to np, provide it to the controller for calculation, and then convert it back to tensor after calculation.

Does the controller have a compatible version of Tensor or a development plan?