3 个章节·按类别展开/折叠
知识
Tensor + Autograd 入门
PyTorch 两大基石:
- Tensor:N 维数组,跟 numpy 互通。
- Autograd:自动求导,requires_grad=True 的 tensor 会记录所有操作,反向传播自动算梯度。
代码
Autograd 示例
知识
关键概念
- .data:绕过 autograd 取原始值
- .detach():从计算图剥离
- with torch.no_grad():推理时不记录梯度,省显存
- torch.cuda.is_available():检查 GPU