运行main.py训练模型报上述错误。其中TF的环境：
Windows10
Anaconda 3
Tensowflow 2.4
Python 3.8
百度了一圈，有几个方法：
1.降低numpy的版本到1.9.5
2.降低Python的版本到3.6
3.修改array_ops.py

个人觉得方法1、2不是真正的解决问题，仅尝试过方法3，可解决此error。

找到array_ops.py，我的在
E:\Anaconda3\envs\tensorflow\Lib\site-packages\tensorflow\python\ops\array_ops.py
根据错误提示，可以找到这个文件位置。
在文件中，增加 from tensorflow.python.ops.math_ops import reduce_prod

 然后找到
def _constant_if_small(value, shape, dtype, name)定义，
原代码是这样的：

 把它修改为：

修改完，运行 main.py，可正常run起来。