![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
PyTorch
长野飘荡
花开花落,人走茶凉
展开
-
PyTorch-线性回归-拟合多项式
先自己定义一个多项式公式,比如:y=2.70∗x3+3.00∗x2+5.00∗x+0.90y = 2.70 * x^3 + 3.00 * x^2 + 5.00 * x + 0.90y=2.70∗x3+3.00∗x2+5.00∗x+0.90然后计算出在 xxx 处于区间 [-3, 3] 时,其对应的 yyy 的值:# 定义多项式参数w = np.array([2.7, 3, 5])b = np.array([0.9])# 生成样本数据x_sample = np.arange(-3, 3, 0.原创 2020-08-22 23:38:32 · 1200 阅读 · 5 评论 -
PyTorch-安装 & 代码自动补全解决方案
pip3 install torch torchvision 安装命令# 构造一个未初始化的5x3矩阵x = torch.empty(5, 3)print(x)# 运行结果tensor([[ 0.0000e+00, 2.0000e+00, 1.3795e+15], [ 8.5920e+09, -1.8056e-34, 4.5745e-41], [...原创 2018-12-26 16:19:56 · 1767 阅读 · 0 评论 -
PyTorch-损失函数-NLLLoss
常用于多分类任务,NLLLoss 函数输入 input 之前,需要对 input 进行 log_softmax 处理,即将 input 转换成概率分布的形式,并且取对数,底数为 eclass torch.nn.NLLLoss(weight=None, size_average=None, ignore_index=-100, reduce=None, reduction='...原创 2019-07-03 15:36:29 · 38592 阅读 · 4 评论 -
PyTorch-损失函数-CrossEntropyLoss
将 input 经过 softmax 激活函数之后,再计算其与 target 的交叉熵损失。阅读源码可以发现,该方法将 log_softmax 和 nll_loss 进行了结合class CrossEntropyLoss(_WeightedLoss): def __init__(self, weight=None, size_average=None, ignore_index=-...原创 2019-07-03 16:49:05 · 7555 阅读 · 0 评论