torch.Size

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

计算值出现的次数。

返回值出现的次数。

返回值首次出现的位置。参数：value（值），start（起始位置，默认为 0），stop（结束位置，默认为 9223372036854775807）。

返回值首次出现的位置。

如果值不存在，则引发 ValueError。

numel() → int

返回一个给定大小的 torch.Tensor 将包含的元素数量。

更正式地说，对于一个大小为 s = torch.Size([10, 10]) 的张量 x = tensor.ones(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 包含的元素数。