I have the following function that used to work, but now when I run it, after upgrading to Tensorflow 2.3, I am getting lots of warnings and dtypes confusing notices. I am not
