【pytorch-tutorial自学笔记】01-basic-PyTorch Basics

1、requires_grad是Pytorch中通用数据结构Tensor的一个属性,用于说明当前量是否需要在计算中保留对应的梯度信息。
在使用神经网络时,全连接层卷积层等结构的参数都是默认需要梯度的。

2、torch.rand() 均匀分布
torch.randn() 正态分布

3、linear = nn.Linear(3, 2) # 构建pytorch中神经网络的线性层,输入为3个特征,输出为2个特征

4、torch.optim.SGD() 参数解释
https://www.cnblogs.com/peixu/p/13194328.html

你可能感兴趣的:(Pytorch,Python,神经网络,深度学习,python)