torch.nn.functional.softmin
-
torch.nn.functional.softmin(input, dim=None, _stacklevel=3, dtype=None)[source] -
Применяет функцию softmin.
Обратите внимание, что . См. определение 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