了解数组的索引

Question

我正在尝试从NG Andrews课程中为MNIST数据集复制神经网络，并且我听不懂关于存储输出值以训练NN将标签记录为矢量的一小段代码。

x.shape = (5000,400)
y.shape = (5000,)
labels = np.unique(y).shape = 10
eye_mat = np.eye(labels)
y_mat = eye_mat[:,y]
y_mat.shape = (10,5000)

对于y_mat的功能，为什么y_mat的形状为（10,5000）以及应该如何存储0到9的输出，我不了解代码的最后一行。任何人都可以简单地将其分解。