Итак, я хочу получить массив с длиной массива индексов, но с vules из массива значений в порядке массива индексов (довольно запутанно), поэтому я хочу получить что-то вот так:
У меня есть массив индексов: [code]test_idxs = np.array([4, 2, 7, 5]) [/code] У меня также есть массив значений (который длиннее): [code]test_vals = np.array([13, 19, 31, 6, 21, 45, 98, 131, 11]) [/code] Итак, я хочу получить массив с длиной массива индексов, но с vules из массива значений в порядке массива индексов (довольно запутанно), поэтому я хочу получить что-то вот так: [code]array([21, 31, 131, 45])[/code] Я знаю, как сделать это в цикле, но не знаю, как добиться этого с помощью инструментов numpy.
У меня проблема с балансировкой строк матрицы CSR. Мне нужно разделить матрицу по строкам на несколько графических процессоров, чтобы каждый графический процессор имел почти одинаковое количество ненулевых элементов.
Для ясности: у меня есть...