神坑啊!!!!我写cs231n的时候,竟然发现了……
binary = np.zeros(p.shape)
print binary.shape
binary[:, y] = 1
print binary
结果如下:
binary = np.zeros(p.shape)
print binary.shape
binary[range(binary.shape[0]), y] = 1
print binary
这才出现了我想要的结果:
Paste_Image.png
多么痛的领悟。。。。再也不敢随便用冒号了!!!