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

傳回值的出現次數。

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

傳回值的首個索引。

如果值不存在，則引發 ValueError。

numel() → int

傳回具有給定大小的 torch.Tensor 所包含的元素數量。

更正式地說，對於一個大小為 s = torch.Size([10, 10]) 的 tensor 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 所包含的元素數量。