Ensure exported models output float tensors

[hugofloresgarcia]

Adds an additional check to register_method to check for output dtypes, ensuring that output tensors are of type float, since the C++ backend will cast all output tensors to float types.