tf.nn.softmax_cross_entropy_with_logits

Posted 马帅领

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tf.nn.softmax_cross_entropy_with_logits相关的知识,希望对你有一定的参考价值。

tf.nn.softmax_cross_entropy_with_logits(logits, labels, name=None)

除去name参数用以指定该操作的name,与方法有关的一共两个参数

 

第一个参数logits:就是神经网络最后一层的输出,如果有batch的话,它的大小就是[batchsize,num_classes],单样本的话,大小就是num_classes

第二个参数labels:实际的标签,大小同上

以上是关于tf.nn.softmax_cross_entropy_with_logits的主要内容,如果未能解决你的问题,请参考以下文章