神坑啊!!!!我写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
这才出现了我想要的结果:

多么痛的领悟。。。。再也不敢随便用冒号了!!!
网友评论