只对输入的张量的最后一个维度做全连接
参考: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)
只对输入的张量的最后一个维度做全连接
参考: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)