class_docstr: Convolutional Neural Network
  with multi-head classifier
code: def forward(self, x, task_labels):
      x = self.features(x)
      x = x.squeeze()
      x = self.classifier(x, task_labels)
      return x
  
file_path: avalanche/avalanche/models/simple_cnn.py
imports: import torch.nn
  from avalanche.models.dynamic_modules import MultiTaskModule, MultiHeadClassifier
label_desc: Use image node_id 2 for calling the MTSimpleCNN obj's underlying member method code with example usage: obj.forward(x, task_labels) and returns: x