一、简述 本篇blog将介绍基于to()的简便的数据类型转换的方法。 很多时候,我们需要将tensor A转换成与tensor B同一类型,这一需求利用to可以简单完成。 to()的另外一种用法是将数据或模型向GPU或者CPU迁移。 二、Demo import torch A = torch.rand(1, 7, 7) B = torch.rand(1, 7, 7) A.to(B)