在PyTorch中，不是所有操作都需要构建计算图。默认情况下，tensor的操作会创建计算图以支持梯度反向传播。withtorch.no_grad()上下文管理器用于阻止计算图的构建，节省内存和计算资源。这对于非训练过程或不需要梯度的场合非常有用。
1.关于with torch.no_grad():
在使用pytorch时，并不是所有的操作都需要进行计算图的生成（计算过程的构建，以便梯度反向传播等操作）。而对于tensor的计算操作，默认是要进行计算图的构建的，在这种情况下，可以使用 with torch.no_grad():，强制之后的内容不进行计算图构建。
————————————————
版权声明：本文为CSDN博主「这是一只小菜鸡」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/weixin_44134757/article/details/105775027
https://www.bmabk.com/index.php/post/12513.html 切分数据集并保留数据集的class和target属性
扫一扫
3209
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
