Function get_onehot

datasets/data_utils.py:147–150

(num_classes, idx)

def get_onehot(num_classes, idx):
    onehot = np.zeros([num_classes], dtype=np.float32)
    onehot[idx] += 1.0
    return onehot