一、问题 TypeError: softmax() got an unexpected keyword argument 'axis' 二、解决办法 TensorFlow版本太低,我的版本为1.4.0 tf.nn.softmax( logits, dim=-1, name=None ) 更新说明:有些参数已被弃用,不推荐使用dim,而是使用axis。 将函数体内axis改成dim即可