tf.keras.losses.CosineSimilarity