Тип int8
Тип данных torch.int8 (или torch.int8) в PyTorch представляет
8-битные целые числа со знаком. Диапазон значений для этого типа
составляет от -128 до 127. Он часто используется для
экономии памяти при работе с большими данными, особенно в задачах,
где точность не критична, например, для хранения меток классов или
в моделях квантования.
При создании тензора тип данных можно указать через параметр
dtype. Также можно преобразовать существующий тензор
в этот тип с помощью метода to или функции type.
Синтаксис
torch.tensor(data, dtype=torch.int8)
Пример
Создадим тензор из списка чисел с типом int8:
import torch
t = torch.tensor([1, 2, 3, 4, 5], dtype=torch.int8)
print(t)
Результат выполнения кода:
tensor([1, 2, 3, 4, 5], dtype=torch.int8)
Пример
Преобразуем существующий тензор в тип int8:
import torch
t = torch.tensor([1.5, 2.7, 3.2])
res = t.to(torch.int8)
print(res)
Результат выполнения кода:
tensor([1, 2, 3], dtype=torch.int8)
При преобразовании числа с плавающей точкой отбрасывается дробная часть.
Пример
Создадим двумерный тензор с типом int8:
import torch
t = torch.tensor([
[1, 2, 3],
[4, 5, 6],
], dtype=torch.int8)
print(t)
Результат выполнения кода:
tensor([
[1, 2, 3],
[4, 5, 6],
], dtype=torch.int8)
