返回学习地图
trainingPhase D · 第 27

D2.1 PyTorch Tensor + 自动求导

PyTorch 两大基石:Tensor 和 Autograd

3 个章节·按类别展开/折叠
知识

Tensor + Autograd 入门

PyTorch 两大基石:

  1. Tensor:N 维数组,跟 numpy 互通。
  2. Autograd:自动求导,requires_grad=True 的 tensor 会记录所有操作,反向传播自动算梯度。
代码

Autograd 示例

知识

关键概念

  • .data:绕过 autograd 取原始值
  • .detach():从计算图剥离
  • with torch.no_grad():推理时不记录梯度,省显存
  • torch.cuda.is_available():检查 GPU