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的主要内容,如果未能解决你的问题,请参考以下文章