Параллелизм с tensorflow

#python #tensorflow #random-forest #decision-tree

#python #tensorflow #случайный лес #дерево решений

Вопрос:

Я пытаюсь распараллелить рост деревьев в случайном лесу с помощью моей собственной реализации дерева решений, но распараллеливание через tensorflow путем построения параллельного графика. tf.map_fn Автоматически ли выполняется функция для каждого подмножества тензора параллельно?

скажите

 tf.map_fn(lambda index: model.fit(xTrain[index,:],yTrain[index,:]).predict(xTest),partitions)
  

Здесь разделы — это список индексов строк, например: [[1,2,3],[2,3,4]]

Также функция имеет аргумент parallel_iterations :

Количество итераций, разрешенных для параллельного выполнения. При построении графика значение по умолчанию равно 10. При быстром выполнении значение по умолчанию устанавливается равным 1

Означает ли аргумент, сколько подмножеств должно выполняться параллельно?

Система, которую я использую, имеет 48 ядер, и я в основном хочу использовать все ядра при создании леса.