TensorFlow 模型建立与训练

感谢博主，问题解决了！修改后的代码在 jupyter notebook 上可以运行了，但是 vscode 依然报一样的错，但也不影响正常运行，我这就去安装 pycharm