Скаляр в PyTorch
Если в тензоре ровно одно число, его называют скаляром. Такой тензор можно превратить в обычное значение Python и передать в обычный код без библиотеки.
Число Python возвращает метод
item. Создадим тензор
из одного дробного значения
и выведем результат:
import torch
weight = torch.tensor(3.5)
print(weight.item()) # выведет 3.5
Тензор из списка с одной ячейкой
тоже содержит одно число, и метод
item сработает так же:
import torch
flag = torch.tensor([1])
print(flag.item()) # выведет 1
Тип результата совпадает с типом
Python: дробный тензор даёт
float, целый - int:
import torch
count = torch.tensor(42)
got = count.item()
print(got) # выведет 42
print(type(got)) # выведет <class 'int'>
Соберите тензор из одного числа
2.5, получите из него
значение Python и выведите
это значение.
Соберите тензор из списка
[0], извлеките число Python
и выведите его.
Соберите тензор из одного целого
17, получите обычное число
и выведите его тип.