TensorFlow DType相关的报错

问题遇到的现象和发生背景 在python环境中导入tensorflow和keras写一维卷积神经网络

问题相关代码，请勿粘贴截图 太多了只能截图了

运行结果及报错内容 TypeError: Cannot convert the argument type_value: torch.float64 to a TensorFlow DType.

我的解答思路和尝试过的方法 怀疑在循环过程中tensorflow的dtype类型发生改变，附上调试的截图

我想要达到的结果