GitHub

tfa.losses.lifted_struct_loss

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 2021-05-17 UTC.

tfa.losses.lifted_struct_loss

Args

Returns