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=(), /)¶
- count(value, /)¶
返回 value 的出现次数。
- index(value, start=0, stop=9223372036854775807, /)¶
返回 value 的第一个索引。
如果 value 不存在,则引发 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
将包含的元素数量。