torch for general PyTorch functionality; torch.nn and torch.nn.functional for neural network based ... correct = top_pred.eq(y.view_as(top_pred)).sum()
確定! 回上一頁