只对输入的张量的最后一个维度做全连接
参考：https://pytorch.org/docs/stable/generated/torch.nn.Linear.html
问题来源
这个output它是个(a,b,c)形式的三维张量，突然做个全连接我不知道咋整，也没见它被展开成一维向量
问题解决
查阅pytorch官方文档后发现，该函数只对输入的张量的最后一个维度做全连接，
即
output(a,b,c)---> nn.Linear(c,d) ---> output(a,c,d)
【d2l】【常见函数】【13】 nn.Linear( )
发布时间 2023-07-27 21:03:42作者: zz子木zz