TensorFlow 升级后变化

作者: 晨箜 | 来源:发表于2017-12-19 14:36 被阅读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的用法有错误,还不知道该怎么改。

4 tf.select

原来:tf.select X

改后:tf.where X

其他参考:
TensorFlow遇到的问题汇总(持续更新中......)

相关文章

网友评论

    本文标题:TensorFlow 升级后变化

    本文链接:https://www.haomeiwen.com/subject/kxgbwxtx.html