我已经导入了tensorflow模块,但无法使用tf.contrib。我不知道问题出在哪里。我尝试在不同的版本中运行,但得到的输出始终相同。
已导入的模块:
import tensorflow.compat.v1 as tf1
tf1.disable_v2_behavior()
import tensorflow as tf2
代码:
tf2.contrib.rnn.LSTMCell(num_units=num_nodes[li],
state_is_tuple=True,
initializer= tf.contrib.layers.xavier_initializer()
)
输出:
AttributeError: module 'tensorflow' has no attribute 'contrib'
回答:
我认为问题出在版本上,我在1.15.2版本中尝试过,它对我来说是有效的。安装上述版本后,尝试以下代码,应该可以工作。
import tensorflow.compat.v1 as tf1
tf1.disable_v2_behavior()
import tensorflow as tf2 #Tensorflow 1.15.2
from tensorflow.contrib.rnn import LSTMCell
LSTMCell(num_units=num_nodes[li],
state_is_tuple=True,
initializer= tf.contrib.layers.xavier_initializer()
)
但是,如果您使用的是TensorFlow 2.x版本,contrib已被弃用,您可以使用以下代码。由于xavier_initializer也使用contrib,您可以使用GlorotUniform初始化器,它与xavier_initializer相同。请按照以下代码操作。
import tensorflow.compat.v1 as tf1
tf1.disable_v2_behavior()
import tensorflow as tf2 #Tensorflow 2.x
tf2.compat.v1.nn.rnn_cell.LSTMCell(num_units=10,
state_is_tuple=True,
initializer= tf2.initializers.GlorotUniform()
)