Torch7卷积神经网络，具有概率的多输出

Question

我使用Torch7进行图像识别（Convolutionnal Neural Network），当我使用默认模型时，即使我有4个输出类，我也会得到一个输出。生成的输出是最终的决定，我想得到每个类（4）的概率，如：

Classe 1 : 75%
Classe 2 : 10%
Classe 3 : 10%
Classe 4 : 5%

以下是我用于定义模型的代码：

-- define model to train
model = nn.Sequential()

-- stage 1 : mean suppresion -> filter bank -> squashing -> max pooling
model:add(nn.SpatialConvolutionMM(1, 32, 5, 5))
model:add(nn.Dropout())
model:add(nn.Tanh())
model:add(nn.SpatialMaxPooling(3, 3, 3, 3))
-- stage 2 : mean suppresion -> filter bank -> squashing -> max pooling
model:add(nn.SpatialConvolutionMM(32, 64, 5, 5))
model:add(nn.Dropout())
model:add(nn.Tanh())
model:add(nn.SpatialMaxPooling(2, 2, 2, 2))
-- stage 3 : standard 2-layer MLP:
model:add(nn.Reshape(64*2*2))
model:add(nn.Linear(64*2*2, 200))
--model:add(nn.Dropout())
model:add(nn.Tanh())
model:add(nn.Linear(200, 4))

感谢您帮助我。