code: def expand_to_planes(x, shape):
      return paddle.tile(append_dims(x, len(shape)), [1, 1, *shape[2:]])
  
imports: import math
  from dataclasses import dataclass
  from functools import partial
  import paddle
  import paddle.nn
label_desc: Use image node_id 2 for calling a global function with example usage: expand_to_planes(x, shape) and returns: paddle