Generalize EnergyInfer expectation to accept other functions

google / qhbm-library

Quantum Hamiltonian-Based Models built on TensorFlow Quantum

https://qhbm-library.readthedocs.io/en/latest/

Apache License 2.0

40 stars 15 forks source link

Generalize EnergyInfer expectation to accept other functions #161

Closed zaqqwerty closed 2 years ago

zaqqwerty commented 2 years ago

Currently the input function maps a batch of bitstrings to a batch of scalars. Can we generalize this to allow: batch of bitstrings to batch of tensors?