Я ищу аналог GPU CuPy для numpy.unique () с опцией оси.
У меня есть 2D-массив Cupy, который мне нужен, чтобы удалить его дублированные строки. К сожалению, функция cupy.unique () выравнивает массив и возвращает одномерный массив с уникальными значениями. Я ищу функцию типа numpy.unique (arr, axis = 0) для решения этой проблемы, но CuPy пока не поддерживает опцию (axis)
x = cp.array([[1,2,3,4], [4,5,6,7], [1,2,3,4], [10,11,12,13]])
y = cp.unique(x)
y_r = np.unique(cp.asnumpy(x), axis=0)
print('The 2D array:\n', x)
print('Required:\n', y_r, 'But using CuPy')
print('The flattened unique array:\n', y)
print('Error producing line:', cp.unique(x, axis=0))
I expect a 2D array with unique rows but I get a 1D array with unique numbers instead. Any ideas about how to implement this with CuPy or numba?