Fix categorical_crossentropy label smoothing for custom axis (#23437)