How to flatten input inside the nn.Sequential? Model = nn.Sequential(x.view(x.shape[0],-1), ... nn.LogSoftmax(dim=1))
確定! 回上一頁