torch.Size

torch.Size是对torch.Tensor.size().它描述了所有维度的大小 原始张量的 Alpha S Alpha 的 Alpha作为tuple，它支持常见的序列作，如 indexing 和 长度。

例：

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

类 Torch 的 Torch 类大小（iterable=（）），/)

count（值，/)

返回 value 的出现次数。

index（值，start=0，stop=9223372036854775807，/)

返回值的第一个索引。

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

numel（） → int

返回元素数 atorch.Tensor将包含 。

更正式地说，对于 size 为 的张量 成立 true。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，而是数字 元素 atorch.Tensor将包含 。