- 博客(2)
- 收藏
- 关注
原创 PyTorch版《动手学深度学习》 2.3自动求梯度
在深度学习中,我们经常需要对函数求梯度(gradient).本节将介绍如何使用PyTorch提供的autograd模块来自动求梯度. from torch import autograd,torch 2.3.1 简单例子 我们先来看一个简单例子:对函数y=2xTxy=2x^Txy=2xTx求关于列向量xxx的梯度.我们先创建变量xxx,并赋初值. x = torch.arange(4).float().reshape(4,1) # 缺失情况下默认 requires_grad = Fa
2022-01-26 13:30:12 2037
原创 Pytorch版《动手学深度学习》 2.2数据操作
2.2.1 创建Tensor 首先导入Pytorch模块. import torch 然后用arange函数创建一个行向量. x = torch.arange(12) x tensor([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]) 通过shape属性获取tensor的形状. x.shape torch.Size([12]) 也可以用size()函数获取tensor的形状. x.size() torch.Size([12]) 下面使用resh
2022-01-24 15:56:26 1427
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人