Warning: This project is deprecated. TensorFlow Addons has stopped development, The project will only be providing minimal maintenance releases until May 2024. See the full announcement here or on github.

tfa.losses.lifted_struct_loss

View source on GitHub

Computes the lifted structured loss.

@tf.function
tfa.losses.lifted_struct_loss(
    labels: tfa.types.TensorLike,
    embeddings: tfa.types.TensorLike,
    margin: tfa.types.FloatTensorLike = 1.0
) -> tf.Tensor

Args
`labels`	1-D tf.int32 `Tensor` with shape `[batch_size]` of multiclass integer labels.
`embeddings`	2-D float `Tensor` of embedding vectors. Embeddings should not be l2 normalized.
`margin`	Float, margin term in the loss definition.

Returns
`lifted_loss`	float scalar with dtype of embeddings.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2023-05-25 UTC.

English
中文 – 简体

tfa.losses.lifted_struct_loss

Args

Returns