你好，这里的评估器将 y_pred 取 argmax，然后和 y_true 的标签比较，可参考 https://tf.wiki/zh_hans/basic/models.html#id27 里的重实现。这是一个手写数字识别的十分类问题，所以输出层的维度必须为 10，以输出模型预测的 “输入数字分别属于 0~9 的概率”，把输出层改为 1 会导致整个模型失去意义。
你好，这里的评估器将 y_pred 取 argmax，然后和 y_true 的标签比较，可参考 https://tf.wiki/zh_hans/basic/models.html#id27 里的重实现。这是一个手写数字识别的十分类问题，所以输出层的维度必须为 10，以输出模型预测的 “输入数字分别属于 0~9 的概率”，把输出层改为 1 会导致整个模型失去意义。