tf.contrib.sparsemax.sparsemax_loss

View source on GitHub

Computes sparsemax loss function 1.

tf.contrib.sparsemax.sparsemax_loss(
    logits,
    sparsemax,
    labels,
    name=None
)

Args:

• logits: A Tensor. Must be one of the following types: half, float32, float64.
• sparsemax: A Tensor. Must have the same type as logits.
• labels: A Tensor. Must have the same type as logits.
• name: A name for the operation (optional).

Returns:

A Tensor. Has the same type as logits.