tf.contrib.layers.one_hot_encoding(
labels,
num_classes,
on_value=1.0,
off_value=0.0,
outputs_collections=None,
scope=None
)
Defined in tensorflow/contrib/layers/python/layers/layers.py
.
Transform numeric labels into onehot_labels using tf.one_hot
.
Args:
labels
: [batch_size] target labels.num_classes
: Total number of classes.on_value
: A scalar defining the on-value.off_value
: A scalar defining the off-value.outputs_collections
: Collection to add the outputs.scope
: Optional scope for name_scope.
Returns:
One-hot encoding of the labels.