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