打开微信,使用扫一扫进入页面后,点击右上角菜单,
点击“发送给朋友”或“分享到朋友圈”完成分享
import torch import torch_mlu.core.mlu_model as ct input1_cpu = torch.ones(2,3) input2_cpu = 2*torch.ones(2,3) input1_mlu = input1_cpu.to(ct.mlu_device()) input2_mlu = input2_cpu.to(ct.mlu_device()) input3_mlu = input1_mlu + input2_mlu input1_mlu = torch.cat((input2_mlu,input1_mlu),0) print(input1_mlu.cpu())
为什么写这样一个简单的程序,调试的时候会出现多个算子无法加载到mlu上的问题,求解
热门帖子
精华帖子