tf.raw_ops.Barrier
Определяет барьер, который сохраняется при разных выполнениях графа.
tf.raw_ops.Barrier(
component_types,
shapes=[],
capacity=-1,
container='',
shared_name='',
name=None
)
Барьер представляет собой отображение ключ-значение, где каждый ключ — строка, а каждое значение — кортеж тензоров.
Во время выполнения барьер содержит элементы «завершенные» и «незавершенные». Завершенный элемент содержит определённые тензоры для всех компонентов кортежа значения и может быть получен с помощью BarrierTakeMany. Незавершенный элемент имеет некоторые неопределённые компоненты в кортеже значения и может быть обновлён с помощью BarrierInsertMany.
| Аргументы | |
|---|---|
component_types | Список tf.DTypes длиной >= 1. Тип каждого компонента в значении. |
shapes | Необязательный список форм (каждая — tf.TensorShape или список ints). По умолчанию []. Форма каждого компонента в значении. Каждая форма должна быть 1 в первом измерении. Длина этого атрибута должна быть такой же, как длина component_types. |
capacity | Необязательный int. По умолчанию -1. Ёмкость барьера. По умолчанию ёмкость — MAX_INT32, которая является максимальной ёмкостью для подлежащей очереди. |
container | Необязательный string. По умолчанию "". Если не пусто, этот барьер размещается в указанном контейнере. В противном случае используется контейнер по умолчанию. |
shared_name | Необязательный string. По умолчанию "". Если не пусто, этот барьер будет совместно использоваться под данным именем в нескольких сессиях. |
name | Название операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Tensor типа mutable string. |
© 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/versions/r2.9/api_docs/python/tf/raw_ops/Barrier