Spec-Zone.ru › PyTorch 2

torch.nn.functional.softmin

torch.nn.functional.softmin(input, dim=None, _stacklevel=3, dtype=None) [source]

Применяет функцию softmin.

Обратите внимание, что Softmin(x)=Softmax(−x)\text{Softmin}(x) = \text{Softmax}(-x). Смотрите определение softmax для математической формулы.

См. Softmin для получения более подробной информации.

Параметры
  • input (Tensor) – входные данные
  • dim (int) – Измерение, по которому будет вычислена softmin (каждый срез по dim будет суммироваться до 1).
  • dtype (torch.dtype, необязательно) – желаемый тип данных возвращаемого тензора. Если указано, тензор входных данных приводится к типу dtype перед выполнением операции. Это полезно для предотвращения переполнения типов данных. По умолчанию: None.
Тип возвращаемого значения

Tensor

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

Spec-Zone.ru

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