torch.Tensor.size¶
- Tensor.size(dim=None) torch.Size or int ¶
返回
self
张量的尺寸。如果未指定dim
,则返回值为torch.Size
,它是tuple
的子类。如果指定了dim
,则返回一个包含该维度尺寸的整数。- 参数
dim (
int
, 可选) – 要检索其尺寸的维度。
示例
>>> t = torch.empty(3, 4, 5) >>> t.size() torch.Size([3, 4, 5]) >>> t.size(dim=1) 4