b = nn.Softmax(dim=0) is used as the softmax function with dim=0 as dimension starts from 0. · input = torch. · output = b(input) is used for ...
確定! 回上一頁