pytorch中数据在CPU与GPU之间的切换
数据在CPU与GPU之间来回切换的pytorch方法：数据从CPU放到GPU，即数据从CPU到GPU的迁移，使用以下语句：data.to("cuda")数据 从GPU到CPU，使用以下语句：data.to("cpu")data通常会有两种数据类型：1. Tensor2. Moduleto函数：转换数据类型/设备1. tensor.to(*args, **kwargs)2. module.to(*args, **kwargs)举例：...
