mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Update docs for torch.mode (#165614)
Currently the docs for `torch.mode` include a note: `This function is not defined for torch.cuda.Tensor yet.` However with `torch==2.7.1+cu126` when I try to get the mode of a Tensor that is in cuda memory, I do not face any issues: ``` >>> a = torch.tensor([0, 2, 1, 1, 1, 3, 3]) >>> a.mode() torch.return_types.mode( values=tensor(1), indices=tensor(4)) >>> a.cuda().mode() torch.return_types.mode( values=tensor(1, device='cuda:0'), indices=tensor(4, device='cuda:0')) ``` Am I misunderstanding the note? If not, I suggest removing it. Pull Request resolved: https://github.com/pytorch/pytorch/pull/165614 Approved by: https://github.com/mikaylagawarecki
This commit is contained in:
committed by
PyTorch MergeBot
parent
9c12651417
commit
a664b299ac
@ -7673,8 +7673,6 @@ If :attr:`keepdim` is ``True``, the output tensors are of the same size as
|
||||
Otherwise, :attr:`dim` is squeezed (see :func:`torch.squeeze`), resulting
|
||||
in the output tensors having 1 fewer dimension than :attr:`input`.
|
||||
|
||||
.. note:: This function is not defined for ``torch.cuda.Tensor`` yet.
|
||||
|
||||
Args:
|
||||
{input}
|
||||
{opt_dim}
|
||||
|
Reference in New Issue
Block a user