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