Download Why use dim=0 when using torch.softmax() for getting predictions ... book pdf free download link or read online here in PDF. Read online Why use dim=0 when using torch.softmax() for getting predictions ... book pdf free download link book now. All books are in clear copy here, and all files are secure so don't worry about it. This site is like a library, you could find million book here by using search box in the header.
It's hard to say why it might be in the tutorial, but usually you have (batch_number, **your_data) shape as input in your network, output in case of classification usually has (batch_number, number_of_classes), and you're right that in that case you should use dim=1(or recommended way use even dim=-1 because you can have more complicated output, for example - (batch_number, some_more_data ...
| Read : Why use dim=0 when using torch.softmax() for getting predictions ... pdf book online Select one of servers for direct link: |
|---|