Метод tolist
Метод tolist преобразует тензор PyTorch в стандартный список Python.
Для скалярных тензоров возвращается число, для одномерных - список чисел,
для многомерных - вложенные списки. Метод не принимает параметров и
всегда создает новый объект списка, не изменяя исходный тензор.
Синтаксис
tensor.tolist()
Пример
Давайте преобразуем одномерный тензор в список чисел:
import torch
t = torch.tensor([1, 2, 3, 4, 5])
res = t.tolist()
print(res)
Результат выполнения кода:
[1, 2, 3, 4, 5]
Пример
Давайте преобразуем двумерный тензор в список списков:
import torch
t = torch.tensor([
[1, 2, 3],
[4, 5, 6],
])
res = t.tolist()
print(res)
Результат выполнения кода:
[[1, 2, 3], [4, 5, 6]]
Пример
Давайте преобразуем скалярный тензор в число:
import torch
t = torch.tensor(42)
res = t.tolist()
print(res)
Результат выполнения кода:
42
Пример
Давайте преобразуем тензор с числами с плавающей точкой:
import torch
t = torch.tensor([1.5, 2.7, 3.1])
res = t.tolist()
print(res)
Результат выполнения кода:
[1.5, 2.7, 3.1]
Пример
Давайте преобразуем трехмерный тензор в глубокий список:
import torch
t = torch.tensor([
[
[1, 2],
[3, 4],
],
[
[5, 6],
[7, 8],
],
])
res = t.tolist()
print(res)
Результат выполнения кода:
[[[1, 2], [3, 4]], [[5, 6], [7, 8]]]
