标签： python numpy
我有一个简单的索引需求，我不知道如何使用numpy来实现它。通常情况下，我们有E的二维数组，其形状为(N,C)，而一维数组y的形状为(N)，这是一个label矩阵，在计算softmax loss时，有一个广泛使用的操作：E[range(N),y]-=1，这是非常有效的代码，现在我有一个更普遍的问题，我有一个shape E的数组{ {1}}并标记形状(C,H,W)的y，现在(H,W)轴是第一个轴，并且C，我如何执行类似3 dimension的操作
E
(N,C)
y
(N)
label
softmax loss
E[range(N),y]-=1
(C,H,W)
(H,W)
C
3 dimension