目录

torch.Size

是对 的调用的结果类型它描述了所有维度的大小 原始张量的 Alpha S Alpha 的 Alpha作为 的子类,它支持常见的序列操作,如 indexing 和 长度。

例:

>>> x = torch.ones(10, 20, 30)
>>> s = x.size()
>>> s
torch.Size([10, 20, 30])
>>> s[1]
20
>>> len(s)
3
Torch 的 Torch 类大小iterable=()),/)
count/)

返回 value 的出现次数。

indexstart=0,stop=9223372036854775807/)

返回值的第一个索引。

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

numel() int

返回具有给定大小的 a 将包含的元素数。

更正式地说,对于 size 为 的张量 成立 true。x = tensor.ones(10, 10)s = torch.Size([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

警告

此函数不返回 描述的维度数,而是返回数字 的元素 a 将包含。

文档

访问 PyTorch 的全面开发人员文档

查看文档

教程

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

查看教程

资源

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

查看资源