Softmin

class torch.nn.Softmin(dim=None) [исходный код]

Применяет функцию Softmin к n-мерному входному тензору.

Изменяет их масштаб так, чтобы элементы n-мерного выходного тензора находились в диапазоне [0, 1] и в сумме равнялись 1.

Softmin определяется следующим образом:

$$\text{Softmin}(x_{i}) = \frac{\exp(-x_i)}{\sum_j \exp(-x_j)}$$

Форма:

• Вход: $(*)$, где * означает любое количество дополнительных измерений
• Выход: $(*)$, той же формы, что и вход

Параметры:

dim (int) – измерение, вдоль которого вычисляется Softmin (таким образом, сумма каждого среза вдоль dim равна 1).

Возвращает:

тензор тех же размерности и формы, что и вход, со значениями в диапазоне [0, 1]

Тип возвращаемого значения:

None

Примеры:

>>> m = nn.Softmin(dim=1)
>>> input = torch.randn(2, 3)
>>> output = m(input)

extra_repr() [исходный код]

Возвращает дополнительное представление модуля.

Тип возвращаемого значения:

str

forward(input) [исходный код]

Выполняет прямой проход.

Тип возвращаемого значения:

Tensor