[bug][#34] remove superfluous squeeze/unsqueeze from this code which led to a
bug in the case where batch_size=1 and n_parameters=1
only in the case of classification parameters is it necessary to remove an extra dimension from the target classes
fixes #34 (closed)