saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[dim order] solve broken doc

Browse source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/146641


ghstack-source-id: 265334899

Differential Revision: [D69265340](https://our.internmc.facebook.com/intern/diff/D69265340/)
This commit is contained in:
gasoonjia 2025-02-07 12:55:57 -08:00
parent 5d7532140f
commit 751c70500c
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
torch/_tensor.py
View file

@ -1517,6 +1517,8 @@ class Tensor(torch._C.TensorBase):
Args:
ambiguity_check (bool or List[torch.memory_format]): The check method for ambiguity of dim order.
Examples::
>>> torch.empty((2, 3, 5, 7)).dim_order()
(0, 1, 2, 3)
>>> torch.empty((2, 3, 5, 7)).transpose(1, 2).dim_order()
@ -1539,6 +1541,7 @@ class Tensor(torch._C.TensorBase):
... except TypeError as e:
... print(e)
The ambiguity_check argument must be a bool or a list of memory formats.
.. warning::
The dim_order tensor API is experimental and subject to change.
"""