code: def infer_shape_numpy_add_sub(fgraph, node, input_shapes):
      ashp, bshp = input_shapes
      # Both inputs should have that same shape, so we just return one of them.
      return [ashp[0]]
  
imports: import unittest
  import pytensor
  from pytensor.graph.basic import Apply
  from pytensor.graph.op import Op
  from pytensor.tensor.type import TensorType
  from tests import unittest_tools
  from pytensor import tensor
  from pytensor.graph.basic import Apply
  from pytensor.tensor.type import dmatrix, matrix
  import numpy
  import pytensor
  from pytensor.compile.ops import as_op
label_desc: Use image node_id 3 for calling a global function with example usage: infer_shape_numpy_add_sub(fgraph, node, input_shapes) and returns: list