One-hot in TF

• tf.one_hot (data to be converted, depth=a few categories)

import tensorflow as tfclasses = 3labels = tf.constant([1, 0, 2])output = tf.one_hot(labels, classes)print(output)

Note: When the variable in the list has decimals, one-hot reports an error