Fixing incorrect doc example in MNASNet. (#3180)

Summary:
* Fixing incorrect doc example in MNASNet.

* Fixing incorrect output.

Reviewed By: fmassa

Differential Revision: D25679208

fbshipit-source-id: 2fc9db36df66104cdf3a29bc514ded93e3131d0f

Author: datumbox

torchvision/models/mnasnet.py

@@ -93,11 +93,11 @@ def _get_depths(alpha: float) -> List[int]:
 class MNASNet(torch.nn.Module):
     """ MNASNet, as described in https://arxiv.org/pdf/1807.11626.pdf. This
     implements the B1 variant of the model.
-    >>> model = MNASNet(1000, 1.0)
+    >>> model = MNASNet(1.0, num_classes=1000)
     >>> x = torch.rand(1, 3, 224, 224)
     >>> y = model(x)
     >>> y.dim()
-    1
+    2
     >>> y.nelement()
     1000
     """