如何快速生成多类别的one-hot向量

Keras

keras 的版本是 2.3.1

>>> import numpy as np
>>> import keras
>>> labels = np.random.randint(10, size=(10, 1))
>>> labels
array([[7],
       [1],
       [8],
       [9],
       [3],
       [1],
       [1],
       [6],
       [1],
       [8]])
>>> one_hot_labels = keras.utils.to_categorical(labels, num_classes=10)
>>> one_hot_labels
array([[0., 0., 0., 0., 0., 0., 0., 1., 0., 0.],
       [0., 1., 0., 0., 0., 0., 0., 0., 0., 0.],
       [0., 0., 0., 0., 0., 0., 0., 0., 1., 0.],
       [0., 0., 0., 0., 0., 0., 0., 0., 0., 1.],
       [0., 0., 0., 1., 0., 0., 0., 0., 0., 0.],
       [0., 1., 0., 0., 0., 0., 0., 0., 0., 0.],
       [0., 1., 0., 0., 0., 0., 0., 0., 0., 0.],
       [0., 0., 0., 0., 0., 0., 1., 0., 0., 0.],
       [0., 1., 0., 0., 0., 0., 0., 0., 0., 0.],
       [0., 0., 0., 0., 0., 0., 0., 0., 1., 0.]], dtype=float32)