如何将元素删除到 mxnet NDArray？

Question

在 numpy 中，可以使用 numpy.delete() 删除数组中的元素。 现在我使用mxnet ndarray 来计算数据，但是我遇到了问题。

如何删除 mxnet ndarray 的元素？

Answer 1

没有从数组中删除单个元素的内置方法，因为通常不需要它。你的情况是什么？为什么要删除元素？

您可以编写自定义代码来执行此操作。如果你有一维数组并且你的代码不会是可混合的，那么这里是如何做到这一点的示例：

import mxnet as mx


def remove_element_by_index(data, index):
    split_data = mx.nd.split(data, num_outputs=data.shape[0], axis=0)
    data_no_element = split_data[:index] + split_data[index + 1:]
    return mx.nd.concat(*data_no_element, dim=0)


data = mx.nd.array([1, 2, 3])

print(remove_element_by_index(data, 0))
print(remove_element_by_index(data, 1))
print(remove_element_by_index(data, 2))

基本上，此代码将数组拆分为多个 1 项数组，然后将其连接回去，而不需要删除。