pytorch 直接申请gpu空间 发表于 2018-08-Sun | 阅读次数: 1mat = torch.zeros((51, 48), dtype=torch.float, device=device)请作者喝一杯咖啡☕️打赏微信支付