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

>>阅读原文<<