Flatten the tensor so it can be fed into the FC layers x = x.view(-1, 320) x = F.relu(self.fc1(x)) x = F.dropout(x, training=self.training)
確定! 回上一頁