tf.nn.softmax_cross_entropy_with_logits(_sentinel=None, labels=None, logits=None,dim=-1, name=None)
参数:_sentinal:一般不用
labels:神经网络期望的输出,它的shape是[batch_size, num_classes]。
logits:神经网络最后一层的输出,如果有batch的话,它的大小就是[batchsize,num_classes],如果是 单样本的话,大小就是
num_classes。(未经缩放,加起来不为1)
name:
操作的名字。
name:
操作的名字。
返回:返回值并不是一个数,而是一个长度为batch_size的张量
。
A 1-D Tensor of length batch_size of the same type as logits with the softmax crossentropy loss.
作用:计算最后一层是softmax层的cross entropy,只不过tensorflow把softmax计算与cross entropy计算放到一起了,用一个函数来实现,用来提高程序的运行速度。