首页 文章

如何在mxnet中计算bincount

提问于
浏览 次
0

我正在尝试使用mxnet为神经网络实现我的客户层 . 我想知道mxnet中是否有类似于np.bincount的函数 . 如果没有，有没有办法我可以计算它而不必将我的 mx.ndarray 转换为 numpy ？

mxnet

1 回答

  • 1

    MXNet没有这样的功能 . 您可以通过编写如下循环来实现它：

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

    请记住，如果你使用numpy，不仅你要从GPU上下文切换到CPU并减慢计算速度，你也不能在计算上做backprop .

    回复于

相关问题