PyTorch
何仙鸟
永远学习
展开
-
Pytorch自动求导机制
PyTorch框架可以帮我们计算好反向传播,需要求导的,可以手动定义。注意:在做反向传播的时候如果不清空会把结果累加起来。可以看到w求导的值跟x一样。原创 2024-01-20 13:14:55 · 631 阅读 · 0 评论 -
PyTorch基本使用方法
可以把tensor格式通过.numpy()转换格式,同时numpy格式也可以通过torch.from_numpy(n)转换格式。x.view(-1,8),-1表示自动计算,计算每列8行会有几列,.size()后会显示[2,8]x.view(16),如果原来矩阵是4*4的那么现在就是1*16的。冒号表示取所有,数字表示取哪一个。原创 2024-01-19 17:27:20 · 358 阅读 · 0 评论 -
PyTorch初识和安装
进入acaconda prompt里面,来到我们官网。Torch:能在GPU中计算的矩阵。原创 2024-01-19 16:51:14 · 300 阅读 · 0 评论