def __init__(self, h_units): super(MLP, self).__init__( l1=L.Linear(784, h_units), l2=L.Linear(h_units, 10) )