torch.Size

Создано: 19 апр. 2024 г. | Последнее обновление: 18 июн. 2025 г.

torch.Size — это тип результата вызова torch.Tensor.size(). Он описывает размеры всех измерений исходного тензора. Будучи подклассом tuple, он поддерживает распространённые операции с последовательностями, такие как индексирование и получение длины.

Пример:

    >>> x = torch.ones(10, 20, 30)
    >>> s = x.size()
    >>> s
    torch.Size([10, 20, 30])
    >>> s[1]
    20
    >>> len(s)
    3

class torch.Size(iterable=(), /)

count(value, /)

Возвращает количество вхождений значения.

index(value, start=0, stop=9223372036854775807, /)

Возвращает первый индекс значения.

Вызывает ValueError, если значение отсутствует.

numel() → int

Возвращает количество элементов, которое содержал бы torch.Tensor с указанным размером.

Формально, для тензора x = tensor.ones(10, 10) с размером s = torch.Size([10, 10]) выполняется x.numel() == x.size().numel() == s.numel() == 100.

Пример:

>>> x=torch.ones(10, 10)
>>> s=x.size()
>>> s
torch.Size([10, 10])
>>> s.numel()
100
>>> x.numel() == s.numel()
True

Предупреждение

Эта функция возвращает не количество измерений, описываемых torch.Size, а количество элементов, которое содержал бы torch.Tensor такого размера.