Как удалить элемент в 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 элемента, а затем объединяет его обратно без того, который нужно удалить.