Guy Moss

FMPE with higher-dim conditions

1

Fixes #1702 This PR addresses the bug for FMPE posterior sampling when the condition event dim is greater than 1. I also added a test which checks that the samples...

FMPE fails to sample with 2D conditions

1

## 🐛 Bug Description When the condition is more than 1-dimensional (e.g. an image instead of a vector), `FMPE` trains successfully but fails to sample because of a shape mismatch....