size = x.size()[1:] # all dimensions except the batch dimension
x.view(-1, x.size()[1:].numel())
numel()函数 直接返回输入的tensor的元素个数/特征空间大小,功能与num_flat_features()等价,还只需要一行代码~,"because its shorter"
x = x.view(x.size()[0], -1)
一般出现在Moduel类的forward函数中,具体位置一般都是在调用分类器之前,在卷积-->激活-->池化之后.
为了将前面操作输出的多维度的tensor展平成一维,然后输入分类器,-1是自适应分配,指在不知道函数有多少列的情况下,根据原tensor数据自动分配列数。