我有一个数据集,我用batch ['data']调用它,并得到我的图像输出MxM。获取图像后,我想用一些numpy操作对其进行处理。在此过程中,我希望我的数据集为我提供带有GPU的图像,然后将输出设备更改为CPU。
我的问题是,Python中函数的隐构化是否按顺序执行?我可以用
base = batch['data'].cuda().cpu()
并且与以下内容相同:
base = batch['data'].cuda()
base.cpu()
提前致谢!
好吧,CPU将完成相同的工作,但结果却不同。
base = batch['data'].cuda().cpu()
在该行之后,您将输出cpu()
存储在名为的变量中base
。
base = batch['data'].cuda()
base.cpu()
在这两行之后,您将输出cuda()
存储在称为的变量中base
,而忘记了的结果cpu()
。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句