torch.Tensor.shape

张量形状

返回self张量的大小。这是size方法的别名。

参见 Tensor.size()

示例:

>>> t = torch.empty(3, 4, 5)
>>> t.size()
torch.Size([3, 4, 5])
>>> t.shape
torch.Size([3, 4, 5])
本页目录