input = input.view(batch_size, -1) # torch.Size([1, 784])# Intialize the linear layer. fc = torch.nn ...
確定! 回上一頁