Метод cpu
Метод cpu переносит тензор на центральный процессор (CPU).
Этот метод полезен, когда вам нужно выполнить операции, которые не поддерживаются
на GPU, либо когда вы хотите сохранить модель или тензор на диске.
Метод возвращает новый тензор, расположенный в памяти CPU, или сам тензор,
если он уже находится на CPU.
Синтаксис
t.cpu()
Пример
Давайте создадим тензор на GPU и перенесём его на CPU:
import torch
t = torch.tensor([1, 2, 3, 4, 5], device='cuda')
res = t.cpu()
print(res)
Результат выполнения кода:
tensor([1, 2, 3, 4, 5])
Пример
Давайте создадим тензор на MPS (Apple Silicon) и перенесём его на CPU:
import torch
t = torch.tensor([1, 2, 3, 4, 5], device='mps')
res = t.cpu()
print(res)
Результат выполнения кода:
tensor([1, 2, 3, 4, 5])
Пример
Метод cpu также работает с двумерными тензорами:
import torch
t = torch.tensor([
[1, 2, 3],
[4, 5, 6],
], device='cuda')
res = t.cpu()
print(res)
Результат выполнения кода:
tensor([
[1, 2, 3],
[4, 5, 6],
])
