Handle broadcasting in DeepEnsemble models

secondmind-labs / trieste

A Bayesian optimization toolbox built on TensorFlow

Apache License 2.0

217 stars 42 forks source link

Closed uri-granta closed 1 year ago

uri-granta commented 1 year ago

The DeepEnsemble.predict method should handle leading batch dimensions, as described in the ProbabilisticModel interface.