tf.compat.v1.nn.softmax_cross_entropy_with_logits_v2 | TensorFlow v2.16.1 (original) (raw)
tf.compat.v1.nn.softmax_cross_entropy_with_logits_v2
Stay organized with collections Save and categorize content based on your preferences.
Computes softmax cross entropy between logits and labels. (deprecated arguments)
tf.compat.v1.nn.softmax_cross_entropy_with_logits_v2(
labels, logits, axis=None, name=None, dim=None
)
Used in the notebooks
| Used in the tutorials |
|---|
| Classify Flowers with Transfer Learning |
Measures the probability error in discrete classification tasks in which the classes are mutually exclusive (each entry is in exactly one class). For example, each CIFAR-10 image is labeled with one and only one label: an image can be a dog or a truck, but not both.
If using exclusive labels (wherein one and only one class is true at a time), see sparse_softmax_cross_entropy_with_logits.
A common use case is to have logits and labels of shape[batch_size, num_classes], but higher dimensions are supported, with the axis argument specifying the class dimension.
logits and labels must have the same dtype (either float16, float32, or float64).
Backpropagation will happen into both logits and labels. To disallow backpropagation into labels, pass label tensors through tf.stop_gradientbefore feeding it to this function.
Note that to avoid confusion, it is required to pass only named arguments to this function.
| Args | |
|---|---|
| labels | Each vector along the class dimension should hold a valid probability distribution e.g. for the case in which labels are of shape[batch_size, num_classes], each row of labels[i] must be a valid probability distribution. |
| logits | Unscaled log probabilities. |
| axis | The class dimension. Defaulted to -1 which is the last dimension. |
| name | A name for the operation (optional). |
| dim | Deprecated alias for axis. |
| Returns |
|---|
| A Tensor that contains the softmax cross entropy loss. Its type is the same as logits and its shape is the same as labels except that it does not have the last dimension of labels. |