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