我想在 Numpy 中实现 Tensorflow 或 PyTorch 的分散和收集操作。我已经挠头好一阵子了。非常感谢任何指点!
最佳答案
有两个内置的 numpy 函数可以满足您的要求。
您可以使用 np.take_along_axis实现 torch.gather,并使用 np.put_along_axis实现 torch.scatter
关于python - 如何在 numpy 中进行分散和聚集操作?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46065873/