Get your layers to fit smoothly, the first time, every time. ... input = torch.randn(1, 28, 28)# Use view() to get [batch_size, ...
確定! 回上一頁