torch.Size#

创建日期：2024年4月19日 | 最后更新日期：2025年6月18日

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=(), /)#

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 所包含的元素总数。

文档