Как рассчитать bincount в mxnet

Question

Я пытаюсь реализовать свой уровень клиента для нейронной сети, используя mxnet.Мне было интересно, если в mxnet есть функция, похожая на np.bincount .Если нет, есть ли способ, которым я могу рассчитать его без необходимости конвертировать mx.ndarray в numpy?

Answer 1

MXNet не имеет такой функциональности. Вы можете реализовать это, написав цикл следующим образом:

bins = []
for i in range(max_value):
    bins.append(nd.sum(my_array == i))
bins = nd.concat(bins)

Имейте в виду, что если вы используете NumPy, вы не только переключаетесь из контекста GPU в CPU и замедляете вычисления, вы также не можете выполнять backprop в вычислениях.