here is the Variable before softmax

屏幕快照 2017-03-31 下午3.15.51.png778×868 83.3 KB


屏幕快照 2017-03-31 下午3.16.01.png712×852 85.9 KB

after softmax, the biggest one of them became 1 and others became 0

屏幕快照 2017-03-31 下午3.16.18.png770×874 71.3 KB

my forward function
def forward(self, x):
    x = F.relu(self.conv1(x))
    x = F.relu(self.conv2(x))
    x = F.relu(self.conv3(x))
    x = F.relu(self.conv4(x))
    x = F.relu(self.conv5(x))
    x = F.relu(self.conv6(x))
    x = F.relu(self.conv7(x))
    x = x.view(x.size(0), -1)
    x = F.softmax(x)
    return x

is there something wrong in my usage of softmax?