reduction=tf.keras.losses.Reduction.NONE) cosine_loss(y_true, y_pred).numpy() array([-0.,-0.999], dtype=float32) compile()API 的用法: model.compile(optimizer='sgd', loss=tf.keras.losses.CosineSimilarity(axis=1))
单机使用: y_true = [[0.,1.], [1.,1.], [1.,1.]] y_pred = [[1.,0.], [1.,1.], [-1.,-1.]] loss = tf.keras.losses.cosine_similarity(y_true, y_pred, axis=1) loss.numpy() array([-0.,-0.999,0.999], dtype=float32)...