tf.raw_ops.DeviceIndex

Возвращает индекс устройства, на котором выполняется операция.

Просмотреть псевдонимы

Псевдонимы для обеспечения совместимости при миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.DeviceIndex

tf.raw_ops.DeviceIndex(
    device_names, name=None
)

Эта операция принимает список имён устройств и возвращает индекс устройства, на котором выполняется операция. Длина списка возвращается в двух случаях: (1) устройство отсутствует в заданном списке устройств; (2) выполняется компиляция XLA.

Аргументы
`device_names`	Список `strings`.
`name`	Имя операции (необязательно).

Возвращает
`Tensor` типа `int32`.

© 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/DeviceIndex