class_docstr: MLPMixer.
  
  Implements
  MLP-Mixer: An all-MLP Architecture for Vision
  https://arxiv.org/abs/2105.01601
code: def input_shape(self) -> torch.Size:
      return torch.Size(self._input_shape)
  
imports: from typing import Tuple, Union
  import torch
  import torch.nn
  from ludwig.utils.torch_utils import LudwigModule
label_desc: Use image node_id 3 for calling the MLPMixer obj's underlying member method code with example usage: obj.input_shape() and returns: torch