torch
文章平均质量分 55
lxznjw
这个作者很懒,什么都没留下…
展开
-
torch小记-常用张量操作
这些代码示例演示了一些常用的张量计算操作,希望能帮助你更好地理解和应用PyTorch中的张量运算。原创 2023-07-12 20:35:43 · 44 阅读 · 0 评论 -
Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
该错误提示 “输入类型(torch.cuda.FloatTensor)和权重类型(torch.FloatTensor)应该相同” 表示输入的张量类型与权重类型不匹配,其中一个张量是在GPU上的cuda张量,而另一个张量是在CPU上的float张量。通过以上方法,您可以确保输入张量和权重张量具有相同的类型和设备,从而避免类型不匹配的错误。请根据您的实际情况选择适当的方法来解决问题。要解决这个问题,您需要确保输入的张量和权重张量具有相同的类型和设备。原创 2023-07-12 10:01:11 · 2541 阅读 · 0 评论 -
Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
简单记录下。这句话翻译一下就是,期望所有张量在同一设备上,但数据分布在两个设备上,cuda:0和cpu。网上基本上就是两种解决方法:方法一、哪一行报错,就把哪一行数据统一下,要么都放再CPU上,要么都放在GPU上。 device = torch.device("cuda" if torch.cuda.is_available() else "cpu") x.to(device)方法二、 x = x.cuda()但是这两种方法..原创 2021-08-14 16:25:17 · 22326 阅读 · 3 评论