tf.raw_ops.TPUReplicatedInput

Подключает N входов к N-стороннему реплицированному вычислению TPU.

Просмотр псевдонимов

Псевдонимы совместимости для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.raw_ops.TPUReplicatedInput

tf.raw_ops.TPUReplicatedInput(
    inputs, is_mirrored_variable=False, index=-1, is_packed=False, name=None
)

Данная операция содержит реплицированный вход в подграф вычислений tpu.replicate(). Каждый реплицированный вход имеет одинаковую форму и тип вместе с выходом.

Например:

%a = "tf.opA"()
%b = "tf.opB"()
%replicated_input = "tf.TPUReplicatedInput"(%a, %b)
%computation = "tf.Computation"(%replicated_input)

Вышеприведенное вычисление имеет реплицированный вход из двух реплик.

Аргументы
`inputs`	Список из как минимум 1 `Tensor` объекта с одинаковым типом.
`is_mirrored_variable`	Необязательный `bool`. По умолчанию `False`.
`index`	Необязательный `int`. По умолчанию `-1`.
`is_packed`	Необязательный `bool`. По умолчанию `False`.
`name`	Имя операции (необязательно).

Возвращаемые значения
`Tensor`. Имеет тот же тип, что и `inputs`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/TPUReplicatedInput