Whether you need a softmax layer to train a neural network in PyTorch will depend on what loss function you use. If you use the torch.nn.
確定! 回上一頁