手写深度学习

Fri, 09 Feb 2024 00:00:00 +0000

与其说深度学习是一门技术，不如说深度学习是一种语言

GitHub 项目地址：AI-Project/scratch

1.1 张量 x 的梯度

张量 \( x \) 的梯度可以存储在 \( x \) 上。

要点：

import torch

# 初始化张量 x (tensor x)
x = torch.arange(4.0)

x.requires_grad_(True)  # 允许 tensor x 存储梯度
x.grad is None  # 梯度默认为 None

> True

初始化带梯度的张量，下面是两个例子：

torch.tensor([1., 2., 3.], requires_grad=True)

> tensor([1., 2., 3.], requires_grad=True)

D2L on Chang Luo