tf.raw_ops.MultiDeviceIterator
Создаёт ресурс MultiDeviceIterator.
tf.raw_ops.MultiDeviceIterator(
devices, shared_name, container, output_types, output_shapes, name=None
)
| Аргументы | |
|---|---|
devices | Список strings длиной >= 1. Список устройств, по которым работает итератор. |
shared_name | string. Если не пусто, этот ресурс будет совместно использоваться под указанным именем в нескольких сессиях. |
container | string. Если не пусто, этот ресурс размещается в заданном контейнере. В противном случае используется контейнер по умолчанию. |
output_types | Список tf.DTypes длиной >= 1. Список типов возвращаемых значений. |
output_shapes | Список форм (каждая — tf.TensorShape или список ints) длиной >= 1. Список генерируемых форм. |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
Tensor типа resource. |
© 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/MultiDeviceIterator