Метод size
Метод size класса Tensor возвращает размер тензора по указанному измерению или кортеж размеров по всем измерениям. Если не передавать аргументов, метод вернёт объект типа torch.Size, содержащий размеры по каждому измерению. Если передать индекс измерения, метод вернёт целое число - длину тензора по данной оси.
Синтаксис
t.size() # возвращает torch.Size
t.size(dim) # возвращает int
Метод принимает один необязательный параметр:
-
dim(int) - индекс измерения, для которого нужно получить размер. Если параметр не указан, метод возвращает размеры по всем осям.
Пример
Давайте создадим двумерный тензор и получим его размеры с помощью метода size:
import torch
t = torch.tensor([
[1, 2, 3],
[4, 5, 6],
])
res = t.size()
print(res)
Результат выполнения кода:
torch.Size([2, 3])
Пример
Давайте получим размер тензора по нулевой оси (количество строк):
import torch
t = torch.tensor([
[1, 2, 3],
[4, 5, 6],
])
res = t.size(0)
print(res)
Результат выполнения кода:
2
В этом примере метод вернул количество строк, так как нулевая ось соответствует строкам двумерного тензора.
Пример
Давайте получим размер тензора по первой оси (количество столбцов):
import torch
t = torch.tensor([
[1, 2, 3],
[4, 5, 6],
])
res = t.size(1)
print(res)
Результат выполнения кода:
3
Пример
Давайте получим размеры трёхмерного тензора:
import torch
t = torch.tensor([
[
[1, 2],
[3, 4],
],
[
[5, 6],
[7, 8],
],
])
res = t.size()
print(res)
Результат выполнения кода:
torch.Size([2, 2, 2])
