人工智能面试题更新 2026-08-05

如何将 tensor 转到指定 device?

北京经纬恒润科技股份有限公司人工智能专业服务技术原理PyTorch

考察说明

考察 PyTorch 中张量设备迁移的基本操作与最佳实践

回答思路

  1. 掌握使用 .to() 方法迁移设备及可能带 dtype 参数
  2. 掌握 .cuda() 与 .cpu() 的用法及适用场景
  3. 理解 device 字符串格式如 'cuda:0'、'cpu'
  4. 说明迁移前数据在 CPU、迁移后显存使用及同步/异步特性
本题已收录答题指导

本题附完整参考答案与评分标准

登录后可查看结构化答题指导;也可以直接开一场模拟面试,AI 面试官用本题实时追问并给出评分。