Устройство в TensorFlow
У тензора есть устройство - блок железа, где лежат его числа и где выполняются операции над ними.
Текущее устройство можно прочитать
через атрибут device. Создадим
ряд и выведем, где он хранится:
import tensorflow as tf
row = tf.constant([1, 2, 3])
print(row.device)
На компьютере без видеокарты тензор лежит на процессоре. Выведем устройство одного числа:
import tensorflow as tf
value = tf.constant(7)
print(value.device)
У таблицы устройство такое же, как у отдельного числа в ней:
import tensorflow as tf
table = tf.constant([[0, 1], [2, 3]])
print(table.device)
Соберите ряд из 4, 5
и 6 и выведите, где
хранятся его данные.
Постройте таблицу из рядов
[1, 2] и [3, 4]
и выведите, где хранятся
её данные.
Соберите тензор из одного числа
100 и выведите, где
лежат его данные.