class_docstr: Multi-layer perceptron with multi-head classifier
class_docstr_tok: ["Multi-layer","perceptron","with","multi-head","classifier"]
code: def forward(self, x, task_labels):
      x = x.contiguous()
      x = x.view(x.size(0), self._input_size)
      x = self.features(x)
      x = self.classifier(x, task_labels)
      return x
  
file_path: avalanche/avalanche/models/simple_mlp.py
imports: import torch.nn
  from avalanche.models.dynamic_modules import MultiTaskModule, MultiHeadClassifier
  from avalanche.models.base_model import BaseModel
label_desc: Use image node_id 2 for calling the MTSimpleMLP obj's underlying member method code with example usage: obj.forward(x, task_labels) and returns: x