【d2l】【常见函数】【13】 nn.Linear( )

只对输入的张量的最后一个维度做全连接
参考：https://pytorch.org/docs/stable/generated/torch.nn.Linear.html

问题来源

image
image

这个output它是个(a,b,c)形式的三维张量，突然做个全连接我不知道咋整，也没见它被展开成一维向量

问题解决

查阅pytorch官方文档后发现，该函数只对输入的张量的最后一个维度做全连接，
即
output(a,b,c)---> nn.Linear(c,d) ---> output(a,c,d)
image
image

posted @ 2023-07-27 20:59  zz子木zz  阅读(74)  评论(0)  编辑  收藏  举报