![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
数产第一混子
授人以鱼,不如授人以渔。
展开
-
torch.gather()函数讲解
torch的gather函数原创 2023-04-20 16:45:29 · 678 阅读 · 0 评论 -
torch.max()的用法
torch.max()的用法。原创 2023-02-19 18:09:56 · 131 阅读 · 0 评论 -
pytorch搭建神经网络
众所周知,训练神经网络的时候,数据一次性全部用来训练效率太慢,所以要分批次来训练数据。自定义神经网络,继承nn.Module类,输入层1个、隐藏层有10个,输出层1个。来更新神经网络的参数值,使得loss达到最小。训练数据,训练轮数为100。原创 2023-02-16 11:28:13 · 609 阅读 · 0 评论 -
torch.squeeze()和torch.unsqueeze()的用法
torch.squeeze()和torch.unsqueeze()的用法原创 2023-02-16 07:00:00 · 239 阅读 · 0 评论