code: def pytorch_model(input_shape: int):
      return torch.nn.Sequential(
          nn.Linear(input_shape, 20), nn.ReLU(), nn.Linear(20, 2)
      )
  
file_path: alibi-detect/alibi_detect/cd/tests/test_classifier.py
imports: import numpy
  import pytest
  import tensorflow
  from tensorflow.keras.layers import Dense, Input
  import torch
  import torch.nn
  from alibi_detect.cd import ClassifierDrift
  from alibi_detect.cd.pytorch.classifier import ClassifierDriftTorch
  from alibi_detect.cd.tensorflow.classifier import ClassifierDriftTF
  from alibi_detect.cd.sklearn.classifier import ClassifierDriftSklearn
  from sklearn.neural_network import MLPClassifier
  from typing import Tuple
label_desc: Use image node_id 2 for calling a global function with example usage: pytorch_model(input_shape) and returns: torch