人工智能面试题更新 2026-08-05
如何将 tensor 转到指定 device?
北京经纬恒润科技股份有限公司人工智能专业服务技术原理PyTorch
考察说明
考察 PyTorch 中张量设备迁移的基本操作与最佳实践
回答思路
- 掌握使用 .to() 方法迁移设备及可能带 dtype 参数
- 掌握 .cuda() 与 .cpu() 的用法及适用场景
- 理解 device 字符串格式如 'cuda:0'、'cpu'
- 说明迁移前数据在 CPU、迁移后显存使用及同步/异步特性
本题附完整参考答案与评分标准
登录后可查看结构化答题指导;也可以直接开一场模拟面试,AI 面试官用本题实时追问并给出评分。