Convert input shape dimension to an integer. (#664)
Convert input shape dimension to an integer. (#664)

* Convert input shape dimension to an integer.

This fixes a case where a user is using Keras and tensorflow.compat.v1, which will send a list of Dimension objects, which do not have a __pow__ function, but do have an __int__ function.

* Update mpo.py

Co-authored-by: Jarrod Kahn <5482d74e2183cb8c0e664f5b49004e1c2816ce44@google.com>
Co-authored-by: Chase Roberts <75a1eb87f49d4aa68dac8f3066fda1a807a88190@google.com>
google/TensorNetwork