Как применить несколько тензорных функций к одному и тому же тензору параллельно? - PullRequest
0 голосов
/ 04 июня 2019

Пусть t - тензор Тензорного потока формы (m, n).Пусть F будет списком p функций f_1, f_2, ..., где каждая из них берет тензор и выдает тензор той же формы.

Мне нужно вычислить тензор T формы (p, m, n) такой, что T[i]=f_i(t).

Обратите внимание, что это обратная настройка функции карты, где мы применяем функцию 1 к коллекции элементов.Здесь нам нужно применить набор функций к 1 элементу.Кроме того, мы должны сделать это в параллельно .

Наивный подход (не параллельный) будет складывать результаты этих вычислений, сгенерированных с помощью цикла for (или понимания списка).

T = tf.stack([
  f(t) for f in F
])

Версия Tensorflow: 2.0.0a.Режим: нетерпеливый

1 Ответ

0 голосов
/ 04 июня 2019

Предполагая, r1.13 без нетерпеливого исполнения tf.parallel_stack должно сработать.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...