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 типа изменяемого 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/api_docs/python/tf/raw_ops/Barrier