def METHOD_NAME(func):
    """Convert the output to a Tensor."""
    @wraps(func)
    def wrapper(*args, **kwargs):
        return bkd.as_tensor(func(*args, **kwargs), dtype=config.real(bkd.lib))
    return wrapper