Encode shape information in `PmOracle`