mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
[docs] torch.min/max: remove incorrect warning from docs (#44615)
Summary: Fixes https://github.com/pytorch/pytorch/issues/44195 cc: mruberry Pull Request resolved: https://github.com/pytorch/pytorch/pull/44615 Reviewed By: ngimel Differential Revision: D23703525 Pulled By: mruberry fbshipit-source-id: 471ebd764be667e29c03a30f3ef341440adc54d2
This commit is contained in:
parent
6bc77f4d35
commit
1d733d660d
|
|
@ -4338,18 +4338,14 @@ value of each row of the :attr:`input` tensor in the given dimension
|
|||
:attr:`dim`. And ``indices`` is the index location of each maximum value found
|
||||
(argmax).
|
||||
|
||||
.. warning::
|
||||
``indices`` does not necessarily contain the first occurrence of each
|
||||
maximal value found, unless it is unique.
|
||||
The exact implementation details are device-specific.
|
||||
Do not expect the same result when run on CPU and GPU in general.
|
||||
For the same reason do not expect the gradients to be deterministic.
|
||||
|
||||
If ``keepdim`` is ``True``, the output tensors are of the same size
|
||||
as ``input`` except in the dimension ``dim`` where they are of size 1.
|
||||
Otherwise, ``dim`` is squeezed (see :func:`torch.squeeze`), resulting
|
||||
in the output tensors having 1 fewer dimension than ``input``.
|
||||
|
||||
.. note:: If there are multiple maximal values in a reduced row then
|
||||
the indices of the first maximal value are returned.
|
||||
|
||||
Args:
|
||||
{input}
|
||||
{dim}
|
||||
|
|
@ -4702,18 +4698,14 @@ value of each row of the :attr:`input` tensor in the given dimension
|
|||
:attr:`dim`. And ``indices`` is the index location of each minimum value found
|
||||
(argmin).
|
||||
|
||||
.. warning::
|
||||
``indices`` does not necessarily contain the first occurrence of each
|
||||
minimal value found, unless it is unique.
|
||||
The exact implementation details are device-specific.
|
||||
Do not expect the same result when run on CPU and GPU in general.
|
||||
For the same reason do not expect the gradients to be deterministic.
|
||||
|
||||
If :attr:`keepdim` is ``True``, the output tensors are of the same size as
|
||||
:attr:`input` except in the dimension :attr:`dim` where they are of size 1.
|
||||
Otherwise, :attr:`dim` is squeezed (see :func:`torch.squeeze`), resulting in
|
||||
the output tensors having 1 fewer dimension than :attr:`input`.
|
||||
|
||||
.. note:: If there are multiple minimal values in a reduced row then
|
||||
the indices of the first minimal value are returned.
|
||||
|
||||
Args:
|
||||
{input}
|
||||
{dim}
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user