Add spatial test for LogSoftmax

Author: desimone

test/common_nn.py

@@ -97,6 +97,12 @@
         input_size=(10, 20),
         reference_fn=lambda i,_: torch.exp(i).div_(torch.exp(i).sum(1).expand(10, 20)).log_()
     ),
+    dict(
+        module_name='LogSoftmax',
+        input_size=(1, 3, 10, 20),
+        reference_fn=lambda i,_: torch.exp(i).div_(torch.exp(i).sum(1).expand_as(i)).log_(),
+        desc='multiparam'
+    ),
     dict(
         module_name='ELU',
         constructor_args=(2.,),