logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

torch.device用法总结

这里写自定义目录标题1 通过字符串构造设备2 通过字符串+序号构造设备3注意事项3.1可直接使用字符串构建3.2 有cuda的也可直接输入序号3.3 以下示例均等效torch.device代表将torch.Tensor分配到的设备的对象。torch.device包含一个设备类型('cpu'或'cuda'设备类型)和可选的设备的序号。如果设备序号不存在,则为当前设备;例如,torch.Tensor用

#pytorch#深度学习#python
到底了