code: def expand_window(window_size: int = 1) -> Model[InT, InT]:
      """For each vector in an input, construct an output vector that contains the
      input and a window of surrounding vectors. This is one step in a convolution.
      """
      return Model(
          "expand_window", forward, attrs={"window_size": window_size}
      )
  
imports: from typing import Callable, Tuple, TypeVar, Union, cast
  from ..config import registry
  from ..model import Model
  from ..types import Floats2d, Ragged
label_desc: Use image node_id 1 for calling a global function with example usage: expand_window(window_size) and returns: Model