Spec-Zone.ru › PyTorch 1

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

Тензор

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

Spec-Zone.ru

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