下面给出 30 道基于 PyTorch 的笔试/面试编程题,每道题都附有详细的代码示例和讲解,涵盖了从基础 Tensor 操作、自动求导,到网络定义、训练流程、数据加载、模型保存、并行计算、混合精度等各个关键点。你可以根据实际需要挑选出最合适的题目进行练习或面试复习。
题目 1:创建 Tensor 及基本操作
问题描述:
请用 PyTorch 创建一个二维 Tensor(例如形状为 [3, 4]
),数据内容由 0~11 的整数构成,并将其转换为浮点数。同时,计算该 Tensor 的转置和均值。
参考答案:
import torch
# 创建一个包含0~11的Tensor并reshape为(3,4)
tensor_int = torch.arange(12).reshape