pytorch学习

tensor.view():

返回一个有相同数据但大小不同的tensor。 返回的tensor必须有与原tensor相同的数据和相同数目的元素，但可以有不同的大小。一个tensor必须是连续的contiguous()才能被查看。

img.size()=torch.Size([128, 1, 28, 28])

img = img.view(num_img, -1)

torch.Size([128, 784])  view操作将img变成了二维，其中第一维是128，第二维由数据自动调整

detach!!!