AttributeError: 模块“tensorflow_core.compat.v1”没有属性“contrib”

Rol*_* B. 4 python jupyter tensorflow tensorflow2.0

x = tf.placeholder(dtype = tf.float32, shape = [None, 28, 28])
y = tf.placeholder(dtype = tf.int32, shape = [None])
images_flat = tf.contrib.layers.flatten(x)
logits = tf.contrib.layers.fully_connected(images_flat, 62, tf.nn.relu)
loss = 
tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits( 
   labels = y, logits = logits))
   train_op = 
   tf.train.AdamOptimizer(learning_rate=0.001).minimize(loss)
   correct_pred = tf.argmax(logits, 1)
   accuracy = tf.reduce_mean(tf.cast(correct_pred, 
   tf.float32))

   print("images_flat: ", images_flat)
   print("logits: ", logits)
   print("loss: ", loss)
   print("predicted_labels: ", correct_pred)


AttributeError                            Traceback (most recent call last)
<ipython-input-17-183722ce66a3> in <module>
      1 x = tf.placeholder(dtype = tf.float32, shape = [None, 28, 28])
      2 y = tf.placeholder(dtype = tf.int32, shape = [None])
----> 3 images_flat = tf.contrib.layers.flatten(x)
      4 logits = tf.contrib.layers.fully_connected(images_flat, 62, tf.nn.relu)
      5 loss = tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits(labels = y, logits = logits))

AttributeError: module 'tensorflow_core.compat.v1' has no attribute 'contrib'
Run Code Online (Sandbox Code Playgroud)

2.这是我在 Jupyter Notebook 中的代码。我刚开始使用 python 并得到我在标题中提到的错误。如果有人可以帮助我提供代码示例来解决问题,我将不胜感激。

Tim*_*lin 9

tf.contrib 曾经在 TensorFlow 2.0 alpha 版本中从 TensorFlow 中删除。

很可能,您已经在使用 TensorFlow 2.0。

您可以在此处找到更多详细信息:https : //github.com/tensorflow/tensorflow/releases/tag/v2.0.0-alpha0

要使用特定版本的 tensorflow,请使用

pip install tensorflow==1.14
Run Code Online (Sandbox Code Playgroud)

或者

pip install tensorflow-gpu==1.14
Run Code Online (Sandbox Code Playgroud)