tf.nn.sparse_softmax_cross_entropy_with_logits(logits=y, labels=tf.argmax(y_,1))
- logits:神经网络不包括softmax层的前向传播结果,是向量格式。
- labels:训练数据的正确答案,是标签格式。可以用tf.argmax(y_,1)来获得正确答案对应的编号。1:在第一维度(每一行)选取最大值对应的下标。
tf.nn.softmax_cross_entropy_with_logits(logits=net, labels=y2)
- labels为one-hot标签