As funções sigmoide e softmax
As funções sigmoide e softmax são funções de ativação essenciais no aprendizado profundo, geralmente usadas como etapa final de uma rede neural.
- A sigmoide serve para classificação binária
- A softmax serve para classificação multiclasse
Dado um tensor de saída de pré-ativação de uma rede, aplique a função de ativação apropriada para obter a saída final.
torch.nn
já foi importado como nn
.
Este exercício faz parte do curso
Introdução ao Aprendizado Profundo com o PyTorch
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
input_tensor = torch.tensor([[2.4]])
# Create a sigmoid function and apply it on input_tensor
sigmoid = nn.____()
probability = ____(____)
print(probability)