Spec-Zone.ru › PyTorch 2

torch.cummin

torch.cummin(input, dim, *, out=None)

Возвращает именованную кортеж (values, indices) , где values представляет собой кумулятивный минимум элементов input по размерности dim. А indices — это индексное расположение каждого максимального значения, найденного по размерности dim.

yi=min(x1,x2,x3,…,xi)y_i = min(x_1, x_2, x_3, \dots, x_i)
Параметры
  • input (Тензор) – входной тензор.
  • dim (int) – размерность для выполнения операции.
Ключевые аргументы

out (кортеж, необязательно) – кортеж результата из двух выходных тензоров (значения, индексы)

Пример:

>>> a = torch.randn(10)
>>> a
tensor([-0.2284, -0.6628,  0.0975,  0.2680, -1.3298, -0.4220, -0.3885,  1.1762,
     0.9165,  1.6684])
>>> torch.cummin(a, dim=0)
torch.return_types.cummin(
    values=tensor([-0.2284, -0.6628, -0.6628, -0.6628, -1.3298, -1.3298, -1.3298, -1.3298,
    -1.3298, -1.3298]),
    indices=tensor([0, 1, 1, 1, 4, 4, 4, 4, 4, 4]))

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.cummin.html

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API