TensorFlow 升级后变化
2017-12-19 本文已影响39人
晨箜
1 调用tf.softmax_cross_entropy_with_logits函数出错。
原来:
tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(y, y_))
改后:
tf.reduce_sum(tf.nn.softmax_cross_entropy_with_logits(logits=y, labels=y_))
2 ‘module’ object has no attribute ‘pack’
原来:tf.pack 、tf.unpack(A, axis)
改后:tf.stack、tf.unstack(A, axis)
解包函数,A需要被解包的对象,axis解包方式(默认=0按行,=1按列)
3 module 'tensorflow.python.ops.nn' has no attribute 'rnn_cell'
原来:
lstm_cell = tf.nn.rnn_cell.BasicLSTMCell(Hidden_size)
cell = tf.nn.rnn_cell.MultiRNNCell([lstm_cell] * Num_layers)
改后:
lstm_cell = tf.contrib.rnn.BasicLSTMCell(Hidden_size)
cell = tf.contrib.rnn.MultiRNNCell([lstm_cell] * Num_layers)
tf.contrib.rnn.MultiRNNCell的用法有错误,还不知道该怎么改。