torch.Size([64]). Now let's take a look at the architecture of our two ... x = torch.flatten(x, 1) x = self.linear_layer(x) x = F.softmax(x, ...
確定! 回上一頁