code: def embedding_inference_rule(
      n: Node, module_instance, symbols, constraints, counter
  ):
      """
      The output shape differs from the input shape in the last dimension
      """
      assert isinstance(n.args[0], Node)
      return gen_embedding_rules(
          n, symbols, module_instance.embedding_dim, counter
      )
  
imports: import torch
  import operator
  import warnings
  from typing import Callable, Dict, Iterable
  from torch.fx._symbolic_trace import _assert_is_none
  from torch.fx.experimental.migrate_gradual_types.constraint import ApplyBroadcasting, CalcProduct, Disj, TGreatestUpperBound, CalcMaxPool, CalcConv, Conj, BinConstraintT, CanReshape, BinConstraintD, GetItem, T, F, TVar, DVar, GetItemTensor, IndexSelect, Transpose, DGreatestUpperBound
  from torch.fx.experimental.migrate_gradual_types.operation import op_eq, op_matching, op_consistency, op_leq, op_precision, op_gt, op_div, op_sub, op_neq, op_lt, op_add, op_mul
  from torch.fx.node import Target, Node
  from torch.fx.experimental.migrate_gradual_types.util import gen_tensor_dims, gen_nat_constraints, gen_dvar, gen_tvar, gen_bvar
  from torch.fx.tensor_type import Dyn, TensorType
  from torch.nn.modules.conv import Conv2d
  from torch.nn.modules.batchnorm import BatchNorm2d
label_desc: Use image node_id 12 for calling a global function with example usage: embedding_inference_rule(n, module_instance, symbols, constraints, counter) and returns: gen_embedding_rules