Python PyTorch Tensor.size用法及代碼示例

本文簡要介紹python語言中 torch.Tensor.size 的用法。

用法:

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

參數：

dim(int,可選的) -要為其檢索大小的維度。

返回self 張量的大小。如果未指定 dim ，則返回值是 torch.Size ，它是 tuple 的子類。如果指定了dim，則返回保存該維度大小的int。

例子：

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