今天在写程序的时候,发现有相应的矩阵形状的不一致发生的报错,具体报错内容如下所示:
从内容之中可以看出来,这个错误是由SparseSoftmaxCrossEntropyWithLogits函数引起的,由于调用的损失函数为keras.losses.SparseCategoricalCrossentropy(from_logits=True),所以想到进入损失函数keras.losses.SparseCategoricalCrossentropy的损失函数之中,去查看相应的交叉熵损失函数的输入和输出
进入到sparse_categorical_crossentropy函数之中,查看相应的函数内容
def sparse_categorical_crossentropy(y_true, y_pred, from_logits=False, axis=-1