tf.raw_ops.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 | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Созданная операция. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/Stage