Python 如何将元素删除到mxnet NDArray？

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

---

如何删除mxnet ndarray的元素？

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

您可以编写自定义代码来实现这一点。下面是一个示例，说明如果您有一维数组，并且您认为您的代码不可混合，那么如何执行此操作：

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项数组，然后将其连接回来，而不需要删除该数组