Устройство в PyTorch

Вычисления над тензором выполняются на устройстве, где лежат его данные. От этого зависит, какой блок железа будет обрабатывать операции.

Текущее устройство можно прочитать через атрибут device. Создадим ряд и выведем, где он хранится:

import torch

row = torch.tensor([1, 2, 3])
print(row.device) # выведет cpu

На этом компьютере новые тензоры обычно появляются на процессоре, и в выводе видна метка процессора:

import torch

value = torch.tensor(7)
print(value.device) # выведет cpu

У таблицы устройство такое же, как у отдельного числа в ней:

import torch

table = torch.tensor([[0, 1], [2, 3]])
print(table.device) # выведет cpu

Соберите ряд из 4, 5 и 6 и выведите, где хранятся его данные.

Соберите таблицу из рядов [1, 2] и [3, 4] и выведите, где хранятся её данные.

Соберите тензор из одного числа 100 и выведите, где лежат его данные.