torch.Tensor.size

Tensor.size(dim=None) → torch.Size or int

回傳 self 張量的尺寸。如果沒有指定 dim，回傳值會是 torch.Size，它是 tuple 的子類別。如果指定了 dim，則會回傳一個整數，表示該維度的尺寸。

參數

dim (int, optional) – 要檢索尺寸的維度。

範例

>>> t = torch.empty(3, 4, 5)
>>> t.size()
torch.Size([3, 4, 5])
>>> t.size(dim=1)
4