Функция PyTorch torch.softmax

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.softmaxэто функция в PyTorch, используемая для вычисления поэлементной функции softmax.

Определение функции

torch.softmax(input, dim, dtype=None)

Пример использования

Пример

import torch

# Вычислить softmax
x = torch.tensor([1.0, 2.0, 3.0])
результат = torch.softmax(x, dim=0)
print(result)

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения