PyTorch
AKA_3
这个作者很懒,什么都没留下…
展开
-
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 阅读 · 0 评论 -
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 阅读 · 0 评论