Linear(512, 784) def forward(self, input): # flatten the input shape = input.shape input = input.view(input.size(0), -1) output_e ...
確定! 回上一頁