np.add.at(a,indices,b)
假设a.shape=[N,T,D],且b.shape = [N,T,D]/[T,D]/[D],则相当于
for indice in indices:
a[indice] += b
一种特殊情况是b.shape=[len(indices),N,T,D],则相当于:
for index,indice in enumerate(indices):
a[indice] += b[index]
np.add.at(a,indices,b)
假设a.shape=[N,T,D],且b.shape = [N,T,D]/[T,D]/[D],则相当于
for indice in indices:
a[indice] += b
一种特殊情况是b.shape=[len(indices),N,T,D],则相当于:
for index,indice in enumerate(indices):
a[indice] += b[index]