Метод cpu
Метод cpu класса Tensor возвращает строку,
содержащую имя устройства CPU, на котором размещен
данный тензор. Метод не принимает параметров и
возвращает значение типа str. Если тензор
размещен на устройстве CPU, метод вернет строку
вида /job:localhost/replica:0/task:0/device:CPU:0.
Метод полезен для определения фактического места
хранения тензора при работе с распределенными
вычислениями и несколькими устройствами.
Синтаксис
t.cpu()
Пример
Давайте создадим тензор и получим имя устройства CPU, на котором он размещен:
import tensorflow as tf
t = tf.constant([1, 2, 3, 4, 5])
res = t.cpu()
print(res)
Результат выполнения кода:
"/job:localhost/replica:0/task:0/device:CPU:0"
Пример
Давайте создадим двумерный тензор и проверим, на каком устройстве CPU он находится:
import tensorflow as tf
t = tf.constant([[1, 2, 3], [4, 5, 6]])
res = t.cpu()
print(res)
Результат выполнения кода:
"/job:localhost/replica:0/task:0/device:CPU:0"
Пример
Давайте создадим тензор строк и получим устройство CPU, на котором он размещен:
import tensorflow as tf
t = tf.constant(['abcde', 'model.keras'])
res = t.cpu()
print(res)
Результат выполнения кода:
"/job:localhost/replica:0/task:0/device:CPU:0"
