如何在`nn.nn.Sequeential'pytorch中平坦的输入

如何在nn.Sequined中弄平输入？

model = nn.Seperention（X.View（X.Shape [0]，-1），

nn.linear（784,256），

nn.relu（），

nn.linear（256,128），

nn.relu（），

nn.linear（128,64），

nn.relu（），

nn.linear（64,10），

nn.logsoftmax（dim = 1））