tf.raw_ops.Stage
Значения Stage, подобные лёгкому Enqueue.
tf.raw_ops.Stage(
values,
capacity=0,
memory_limit=0,
container='',
shared_name='',
name=None
)
Основная функциональность данного Оператора похожа на очередь с намного меньшими возможностями и вариантами. Данный оператор оптимизирован для производительности.
| Аргументы | |
|---|---|
values | Список объектов Tensor. Список типов данных, которым должны соответствовать вставляемые значения. |
capacity | Необязательный int, который является >= 0. По умолчанию 0. Максимальное количество элементов в области размещения. Если > 0, вставки в контейнер будут блокироваться, когда емкость будет достигнута. |
memory_limit | Необязательный int, который является >= 0. По умолчанию 0. Максимальное количество байтов, разрешенных для тензоров в области размещения. Если > 0, вставки будут блокироваться до тех пор, пока не будет доступно достаточно места. |
container | Необязательный string. По умолчанию "". Если не пусто, данная очередь размещается в указанном контейнере. В противном случае используется контейнер по умолчанию. |
shared_name | Необязательный string. По умолчанию "". Необходимо согласовать это имя с соответствующим оператором Unstage. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Созданная операция. |
© 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/Stage