张量的创建默认是在CPU上,所以可以加to(device="cuda:number"),自动在GPU上创建张量
张量创建tip
最新推荐文章于 2024-06-07 10:10:42 发布
张量的创建默认是在CPU上,所以可以加to(device="cuda:number"),自动在GPU上创建张量