Added missing import to TwoLayerMLP

bob/learn/pytorch/architectures/TwoLayerMLP.py

@@ -8,6 +8,9 @@
 
 from torch import nn
 
+import torch.nn.functional as F
+
+
 #==============================================================================
 # Define the network:
 
@@ -22,7 +25,7 @@ class TwoLayerMLP(nn.Module):
         Dimensionality of the input feature vectors.
 
     n_hidden_relu : int
-        Number of ReLU units in the hidden layer of the MLP
+        Number of ReLU units in the hidden layer of the MLP.
     """
 
     def __init__(self, in_features, n_hidden_relu):