自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(2)
  • 收藏
  • 关注

原创 Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu

device = torch.device('cuda') torch.rand(8,1).to(device) 原因在于torch函数生成的tensor需要转到GPU上

2022-03-29 09:55:55 1698

原创 Pytorch:均匀分布(a,b)[tensor类型]

(b-a)*torch.rand(m,n)+a [m,n]:生成均匀分布的形状

2022-03-29 09:42:40 2875

空空如也

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除