Метод element_size
Метод element_size возвращает размер одного элемента тензора в байтах.
Этот метод не принимает никаких параметров и возвращает целое число,
которое зависит от типа данных элементов тензора.
Например, для типа torch.float32 размер элемента будет 4 байта,
а для torch.float64 - 8 байт.
Синтаксис
tensor.element_size()
Пример
Давайте создадим тензор с типом torch.float32 и проверим размер его элементов:
import torch
t = torch.tensor([1, 2, 3, 4, 5], dtype=torch.float32)
res = t.element_size()
print(res)
Результат выполнения кода:
4
Пример
Теперь создадим тензор с типом torch.float64:
import torch
t = torch.tensor([1, 2, 3, 4, 5], dtype=torch.float64)
res = t.element_size()
print(res)
Результат выполнения кода:
8
Пример
Рассмотрим размеры элементов для разных целочисленных типов:
import torch
t_int8 = torch.tensor([1, 2, 3], dtype=torch.int8)
t_int16 = torch.tensor([1, 2, 3], dtype=torch.int16)
t_int32 = torch.tensor([1, 2, 3], dtype=torch.int32)
t_int64 = torch.tensor([1, 2, 3], dtype=torch.int64)
print(t_int8.element_size())
print(t_int16.element_size())
print(t_int32.element_size())
print(t_int64.element_size())
Результат выполнения кода:
1
2
4
8
Пример
Метод element_size можно использовать вместе с numel,
чтобы вычислить общий размер тензора в байтах:
import torch
t = torch.ones(100, 100, dtype=torch.float64)
total_bytes = t.element_size() * t.numel()
print(total_bytes)
Результат выполнения кода:
80000
