img.view(img.size(0), -1)
时间: 2023-10-20 10:56:53 浏览: 76
img.view(img.size(0), -1)是将图片展开成一维向量的操作。在深度学习中,通常将图片表示为二维矩阵,其中每个元素代表像素的强度值。而在某些情况下,我们需要将图片展平成一维向量,以便输入到神经网络中进行处理。这个操作可以通过调用PyTorch中的view函数来实现,其中img是输入的图片张量,img.size(0)表示图片的批次大小,-1表示自动计算展平后的长度。这样,img.view(img.size(0), -1)就可以将图片展开成一维向量。\[1\]\[2\]
#### 引用[.reference_title]
- *1* *2* [pytorch MNIST study(一)](https://blog.csdn.net/qq_42527487/article/details/84833494)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
- *3* [Pytorch基于全连接神经网络训练MNIST之入门篇](https://blog.csdn.net/xjm850552586/article/details/109171016)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]