У меня есть неотрицательный нормализованный вектор p. Я хотел бы выбрать индекс из набора индексов вектора. Вероятность получения выборки k равна p[k]. Используя функцию np.random.choise, я могу написать следующий код.
Код: Выделить всё
p = [0.2, 0.3, 0.1, 0.3, 0.1]
indices = np.arange(len(p))
k = np.random.choice(indices, p=p)
Мой вопрос: как я могу обобщить этот код для многомерных массивов? Например, учитывая трехмерный неотрицательный нормализованный тензор IxJxK p = np.random.rand(I,J,K)
как я могу выполнить выборку индекса (i,j,k) с переносимостью p[i,j,k]?
Подробнее здесь:
https://stackoverflow.com/questions/791 ... -in-python
Мобильная версия