mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
58e403c739
Fixes #61231. Fixes #124167. This PR documents a rather long-standing issue w.r.t. unexpected behavior of `torch.Size.numel`, first reported almost 5 years ago. Pull Request resolved: https://github.com/pytorch/pytorch/pull/124186 Approved by: https://github.com/janeyx99
26 lines
522 B
ReStructuredText
26 lines
522 B
ReStructuredText
torch.Size
|
|
===================================
|
|
|
|
:class:`torch.Size` is the result type of a call to :func:`torch.Tensor.size`. It describes the size of all dimensions
|
|
of the original tensor. As a subclass of :class:`tuple`, it supports common sequence operations like indexing and
|
|
length.
|
|
|
|
|
|
Example::
|
|
|
|
>>> x = torch.ones(10, 20, 30)
|
|
>>> s = x.size()
|
|
>>> s
|
|
torch.Size([10, 20, 30])
|
|
>>> s[1]
|
|
20
|
|
>>> len(s)
|
|
3
|
|
|
|
|
|
|
|
.. autoclass:: torch.Size
|
|
:members:
|
|
:undoc-members:
|
|
:inherited-members:
|