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