如何使IDEA / PyCharm正确对待@tf_export?

时间:2019-03-16 22:07:51

标签: python tensorflow intellij-idea pycharm tf.keras

如何使IDEA / PyCharm正确对待@tf_export批注?

我切换到TF 1.12和以下代码

import tensorflow as tf
from tensorflow.keras import layers

model = tf.keras.Sequential()
model.add(layers.Dense(64, activation='relu'))

我看到IDEA突出显示了layers的导入。

enter image description here

我无法通过右键单击来声明Dense的声明。

代码正确执行。

有没有办法解决这个问题?

1 个答案:

答案 0 :(得分:0)

您可以这样做:

import tensorflow.python.keras
from keras import Sequential, layers