TypeError: softmax() got an unexpected keyword argument 'axis'

一、问题

TypeError: softmax() got an unexpected keyword argument 'axis'

二、解决办法

TensorFlow版本太低，我的版本为1.4.0

tf.nn.softmax(
    logits,
    dim=-1,
    name=None
)

更新说明：有些参数已被弃用，不推荐使用dim，而是使用axis。

将函数体内axis改成dim即可