Función torch.softmax de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.softmaxEs una función de PyTorch que se utiliza para calcular la función softmax elemento a elemento.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

# Calcular softmax
x = torch.tensor([1.0, 2.0, 3.0])
result = torch.softmax(x, dim=0)
print(result)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones