tf.distribute.CrossDeviceOps
Базовый класс для алгоритмов кросс-устройства для сокращения и трансляции.
tf.distribute.CrossDeviceOps()
Методы
batch_reduce
Просмотреть исходный код
batch_reduce(
reduce_op, value_destination_pairs
)
Сведение PerReplica объектов в пакет.
Сведение каждого первого элемента в value_destination_pairs к каждому второму элементу, который указывает назначения.
| Аргументы |
|---|
reduce_op | Указывает, как будет производиться сокращение per_replica_value. Допустимые значения — tf.distribute.ReduceOp.SUM, tf.distribute.ReduceOp.MEAN. |
value_destination_pairs | список или кортеж кортежей объектов PerReplica (или тензоров с заданным устройством, если устройство одно) и назначений. |
| Возвращаемое значение |
|---|
| список объектов Mirrored. |
| Исключения |
|---|
ValueError | если value_destination_pairs не является списком или кортежем кортежей объектов PerReplica и назначений |
batch_reduce_implementation
Просмотреть исходный код
batch_reduce_implementation(
reduce_op, value_destination_pairs
)
Реализация сведения PerReplica объектов в пакет.
Сведение каждого первого элемента в value_destination_pairs к каждому второму элементу, который указывает назначения.
| Аргументы |
|---|
reduce_op | Указывает, как будет производиться сокращение per_replica_value. Допустимые значения — tf.distribute.ReduceOp.SUM, tf.distribute.ReduceOp.MEAN. |
value_destination_pairs | список или кортеж кортежей объектов PerReplica (или тензоров с заданным устройством, если устройство одно) и назначений. |
| Возвращаемое значение |
|---|
| список объектов Mirrored. |
| Исключения |
|---|
ValueError | если value_destination_pairs не является списком или кортежем кортежей объектов PerReplica и назначений |
broadcast
Просмотреть исходный код
broadcast(
tensor, destinations
)
Трансляция tensor к назначениям.
| Аргументы |
|---|
tensor | тензор для трансляции. |
destinations | назначения для трансляции. |
| Возвращаемое значение |
|---|
| объект Mirrored. |
broadcast_implementation
Просмотреть исходный код
broadcast_implementation(
tensor, destinations
)
Реализация трансляции tensor к назначениям.
| Аргументы |
|---|
tensor | тензор для трансляции. |
destinations | назначения для трансляции. |
| Возвращаемое значение |
|---|
| объект Mirrored. |
reduce
Просмотреть исходный код
reduce(
reduce_op, per_replica_value, destinations
)
Сведение per_replica_value к destinations.
Выполняет операцию сведения, определенную reduce_op, и помещает результат в destinations.
| Аргументы |
|---|
reduce_op | Указывает, как будет производиться сокращение per_replica_value. Допустимые значения — tf.distribute.ReduceOp.SUM, tf.distribute.ReduceOp.MEAN. |
per_replica_value | объект PerReplica или тензор с заданным устройством. |
destinations | назначения для сведения. |
| Возвращаемое значение |
|---|
| объект Mirrored. |
| Исключения |
|---|
ValueError | если per_replica_value нельзя преобразовать в объект PerReplica. |
reduce_implementation
Просмотреть исходный код
reduce_implementation(
reduce_op, per_replica_value, destinations
)
Реализация сведения per_replica_value к destinations.
Выполняет операцию сведения, определенную reduce_op, и помещает результат в destinations.
| Аргументы |
|---|
reduce_op | Указывает, как будет производиться сокращение per_replica_value. Допустимые значения — tf.distribute.ReduceOp.SUM, tf.distribute.ReduceOp.MEAN. |
per_replica_value | объект PerReplica или тензор с заданным устройством. |
destinations | назначения для сведения. |
| Возвращаемое значение |
|---|
| объект Mirrored. |
| Исключения |
|---|
ValueError | если per_replica_value нельзя преобразовать в объект PerReplica. |
