Patch QNN _expectation_function and add associated test.
Since the expectation and pulled_back_expectation functions expect operators to be 1D, needed to add tiling of operators to the underlying _expectation_function body. Also took the opportunity to add a test: to make sure it works with both reduce == True and reduce == False, as well as testing the gradients.
Note I had to remove the @tf.function decorators, something goes wrong with TFQ when gradients are called on the decorated function. Opened issue #71 to track resolution of this.
Patch QNN
_expectation_functionand add associated test.Since the
expectationandpulled_back_expectationfunctions expectoperatorsto be 1D, needed to add tiling of operators to the underlying_expectation_functionbody. Also took the opportunity to add a test: to make sure it works with bothreduce == Trueandreduce == False, as well as testing the gradients.Note I had to remove the
@tf.functiondecorators, something goes wrong with TFQ when gradients are called on the decorated function. Opened issue #71 to track resolution of this.