tf.contrib.tpu.outfeed_dequeue
Извлекает один тензор из вычислительного выходного буфера.
tf.contrib.tpu.outfeed_dequeue(
dtype, shape, device_ordinal=-1, name=None
)
Эта операция будет заблокирована до тех пор, пока данные не станут доступными.
| Аргументы | |
|---|---|
dtype | tf.DType. Тип элементов в тензоре. |
shape | tf.TensorShape или список ints. Форма тензора. |
device_ordinal | Необязательный int. По умолчанию -1. TPU-устройство, которое нужно использовать. Должно быть -1, когда операция выполняется на TPU-устройстве, и >= 0, когда операция выполняется на CPU-устройстве. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
Tensor типа dtype. |
© 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/r1.15/api_docs/python/tf/contrib/tpu/outfeed_dequeue
