TensorFlow tf.keras.losses.SparseCategoricalCrossentropy

下面的例子,y_true的形状是[batch_size],y_pred的形状是[batch_size,num_class]. 也就是说y_pred通过softmax层以后,是one-hot编码,SparseCategoricalCrossentropy将y_true变成了one-hot编码.python cce = tf.keras.losses.SparseCategoricalCrossent

>>阅读原文<<