Атрибут device
Атрибут device класса Tensor возвращает строку,
содержащую полное имя устройства, на котором хранится тензор.
Это может быть центральный процессор (CPU) или графический
процессор (GPU). Атрибут доступен только для чтения и не
принимает параметров.
Синтаксис
t.device
Пример
Давайте создадим тензор и посмотрим, на каком устройстве он размещен по умолчанию:
import tensorflow as tf
t = tf.constant([1, 2, 3, 4, 5])
print(t.device)
Результат выполнения кода:
"/job:localhost/replica:0/task:0/device:CPU:0"
Пример
Давайте явно разместим тензор на CPU с помощью контекстного
менеджера tf.device и проверим атрибут device:
import tensorflow as tf
with tf.device('/CPU:0'):
t = tf.constant([1, 2, 3, 4, 5])
print(t.device)
Результат выполнения кода:
"/job:localhost/replica:0/task:0/device:CPU:0"
Пример
Давайте создадим двумерный тензор и выведем его устройство вместе с другими атрибутами:
import tensorflow as tf
t = tf.constant([[1, 2, 3], [4, 5, 6]])
print("Device:", t.device)
print("Shape:", t.shape)
print("Dtype:", t.dtype)
Результат выполнения кода:
Device: /job:localhost/replica:0/task:0/device:CPU:0
Shape: (2, 3)
Dtype: <dtype: 'int32'>
