pyrorch python out.view(out.size(0), -1) out.view(-1, 1, 28, 28) clamp(min,max)
1.view(out.size(0),-1)目的是将多维的的数据如(none,36,2,2)平铺为一维如(none,144)。作用类似于keras中的Flatten函数。只不过keras中是和卷积一起写的,而pytorch是在forward中才声明的。defforward(self,x):out=self.conv(x)out=out.view(out.size(0),-1)out=self.fc