size.rst

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: