用于三维多层感知的torch代码

Question

我想把数据A (100,196,512)转换成数据B(100，1,512)。这看起来像是一种多层次的感知。但是，函数nn.Linear不能用于三维。如何解决这个问题？

谢谢!

Answer 1

所以从本质上讲，你想把一批尺寸为196的100*512个向量通过一个网络。要做到这一点，你需要将问题本质上重述为一批1D问题，例如

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模块，这会更容易。