Aproveite o mês das carreiras na alura
Até 44% OFF
0 dias
0 horas
0 min
0 seg
Referente ao curso Treinando uma Rede Neural: Deep Learning com PyTorch, no capítulo Funções de Perda e atividade Regressão
Por que foi utilizada a dim = -1 na função softmax()?
Mês das carreiras alura
Aproveite o lançamento das carreiras e comece a estudar com descontos!
A dimensão do dado que flui na rede é B X N (tamanho do batch x tamanho da feature). O índice -1 se refere à última dimensão, ou seja, a dimensão da feature.