TPUReplicatedOutput

публичный финальный класс TPUReplicationOutput

Соединяет N выходов N-путевого реплицированного вычисления TPU.

Эта операция хранит реплицированные выходные данные из подграфа вычислений tpu.rescribe(). Каждый реплицированный выходной сигнал имеет ту же форму и тип, что и входной.

Например:

%computation = "tf.Computation"()
 %replicated_output:2 = "tf.TPUReplicatedOutput"(%computation)
 
Приведенное выше вычисление имеет реплицированные выходные данные двух реплик.

Публичные методы

статический <T> TPUReplicationOutput <T>
create (область действия , ввод операнда <T>, длинные числа реплик)
Фабричный метод для создания класса, обертывающего новую операцию TPUReplicationOutput.
Итератор< Операнд <T>>
Список< Выход <T>>

Унаследованные методы

Публичные методы

public static TPUReplicationOutput <T> create (область действия , ввод операнда <T>, длинное число реплик)

Фабричный метод для создания класса, обертывающего новую операцию TPUReplicationOutput.

Параметры
объем текущий объем
Возврат
  • новый экземпляр TPUReplicationOutput

общедоступный итератор < Операнд <T>> итератор ()

публичный список< Вывод <T>> выходы ()