Merge pull request tensorflow#34625 from anuj-rawat:unittest_fix_avx512

PiperOrigin-RevId: 294730950
Change-Id: I7cb7483b6865217b0230873d94c598f055c09496

Author: tensorflower-gardener

tensorflow/python/keras/backend.py

@@ -4547,12 +4547,11 @@ def categorical_crossentropy(target, output, from_logits=False, axis=-1):
      [0.5  0.89 0.6 ]
      [0.05 0.01 0.94]], shape=(3, 3), dtype=float32)
   >>> loss = tf.keras.backend.categorical_crossentropy(a, b)
-  >>> print(loss)
-  tf.Tensor([0.10536055 0.8046684  0.06187541], shape=(3,), dtype=float32)
+  >>> print(np.around(loss, 5))
+  [0.10536 0.80467 0.06188]
   >>> loss = tf.keras.backend.categorical_crossentropy(a, a)
-  >>> print(loss)
-  tf.Tensor([1.1920929e-07 1.1920929e-07 1.1920930e-07], shape=(3,),
-  dtype=float32)
+  >>> print(np.around(loss, 5))
+  [0. 0. 0.]
 
   """
   target.shape.assert_is_compatible_with(output.shape)