AKA_3-CSDN博客

原创 PyTorch的AutoGrad

标量自动求导输入import torchx = torch.arange(4.0)x.requires_grad_(True) # 等价于 `x = torch.arange(4.0, requires_grad=True)`x.grad # 默认值是Noney = 2 * torch.dot(x, x)y.backward()x.grad输出tensor([ 0., 4., 8., 12.])非标量变量的反向传播输入# 对非标量调用`backward`需要传入一个`

2021-06-17 12:18:18 128

原创 PyTorch线性代数方法

标量构造x = torch.tensor([3.0])向量构造输入：x = torch.arange(4)x输出：tensor([0, 1, 2, 3])点乘输入y = torch.ones(4, dtype=torch.float32)x, y, torch.dot(x, y)输出(tensor([0., 1., 2., 3.]), tensor([1., 1., 1., 1.]), tensor(6.))矩阵构造输入A = torch.arange(20)

2021-06-14 11:57:32 335

原创 Vanilla SGD

朴素随机梯度下降（Vanilla SGD）通过将当前权重减去其梯度的因子（即，学习率）来更新权重。

2021-06-10 19:11:20 883

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人