code: def _run(self, theta, size, align_corners=None):  # type: ignore
      align_corners = align_corners or self.align_corners  # type: ignore
      _, _, *data_size = size
      original_grid = construct_original_grid(data_size, align_corners)
      grid = apply_affine_transform(theta, original_grid)
      return (grid,)
  
imports: import numpy
  from onnx.reference.op_run import OpRun
label_desc: Use image node_id 1 for calling the AffineGrid obj's underlying member method code with example usage: obj._run(theta, size, align_corners) and returns: grid