baybe.surrogates.utils.batchify_mean_var_prediction¶ baybe.surrogates.utils.batchify_mean_var_prediction(posterior: Callable[[_TSurrogate, Tensor], tuple[Tensor, Tensor]])[source]¶ Wrap a posterior method to make it evaluate t-batches as an augmented q-batch. Return type: Callable[[TypeVar(_TSurrogate, bound= Surrogate), Tensor], tuple[Tensor, Tensor]]