![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
有点聪明的亚子1
这个作者很懒,什么都没留下…
展开
-
torch.stack()改变维度的一种用法
import torchc = []b = torch.Tensor([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])# stack()函数要求输入为Tensors,而不是Tensor,这里放入两次tensorc.append(b)c.append(b)# print(c.shape) # [2, 3, 4]a = torch.stack(c, dim=1)print(a.shape) # [3, 2, 4]...原创 2021-12-09 16:10:50 · 1198 阅读 · 0 评论 -
nn.Linear的个人理解
PyTorch的nn.Linear()是用于设置网络中的全连接层的。输入张量维度:(a, c)想要的输出张量维度:(a,d)那么就应该这样写:nn.Linear(c, d)全连接层也就是做矩阵相乘,这里相当于构建了一个矩阵维度是(c, d)那么(a, c)*(c, d) =(a, d)在分类问题中用作最后一层改变分类的个数。...原创 2021-09-25 20:09:17 · 2467 阅读 · 2 评论