用于三维多层感知的火炬代码

Question

我想将数据A（100,196,512）传输到数据B（100,1,512）。 这似乎是一个多层感知。但是，函数nn.Linear不能用于三维。如何解决这个问题？

谢谢！

Answer 1

所以基本上你想要通过网络放置一批100×512维度196的向量。为此，您需要将问题重新解决为一批一维问题，例如

model = nn.Sequential()
model:add( nn.Transpose({2},{3}) )
model:add( nn.View(100*512,196) )
model:add( nn.Linear(196,1) )
model:add( nn.View(100,512,196) )
model:add( nn.Transpose({2},{3}) )

如果您的数据是A（100,512,196），只需要两个nn.View模块，这将更容易。