np.eye()的用法
numpy.eye(N,M,k)
N:行数
M:列数
k默认为0,表示主对角线为1,其他为0,若为正则上对角线为1,若为负则下对角线为1
one-hot用法
a=np.eye(3)[1]
得到:[0. 1. 0.]
a=np.eye(3)[2]
得到:[0. 0. 1.]
labels=np.array([[1],[2],[0],[1]])
res=np.eye(3)[labels.reshape(-1)
[[0. 1. 0.]
[0. 0. 1.]
[1. 0. 0.]
[0. 1. 0.]]