快捷方式

torch.Size

torch.Size 是调用 torch.Tensor.size() 的结果类型。它描述了原始张量所有维度的尺寸。作为 tuple 的子类,它支持常见的序列操作,如索引和长度。

示例

>>> x = torch.ones(10, 20, 30)
>>> s = x.size()
>>> s
torch.Size([10, 20, 30])
>>> s[1]
20
>>> len(s)
3
class torch.Size(iterable=(), /)
count(value, /)

返回 value 的出现次数。

index(value, start=0, stop=9223372036854775807, /)

返回 value 的第一个索引。

如果 value 不存在,则引发 ValueError。

numel() int

返回具有给定尺寸的 torch.Tensor 将包含的元素数量。

更正式地说,对于具有尺寸 s = torch.Size([10, 10]) 的张量 x = tensor.ones(10, 10)x.numel() == x.size().numel() == s.numel() == 100 为真。

示例:
>>> x=torch.ones(10, 10)
>>> s=x.size()
>>> s
torch.Size([10, 10])
>>> s.numel()
100
>>> x.numel() == s.numel()
True

警告

此函数不返回由 torch.Size 描述的维度数量,而是返回具有该尺寸的 torch.Tensor 将包含的元素数量。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源