code: def __init__(self, dim, shape):
      super().__init__()
  
      params = paddle.randn([1, dim, shape])
      self.pos_embed = self.create_parameter(
          shape=params.shape,
          dtype=str(params.numpy().dtype),
          default_initializer=nn.initializer.Assign(params),
      )
      self.pos_embed.stop_gradient = False
  
imports: import paddle
  import paddle.nn
  import paddle.nn.functional
  from paddleseg.cvlibs import manager, param_init
  from paddleseg.models import layers
  from paddleseg.models.backbones.transformer_utils import DropPath, zeros_, ones_
  from paddleseg.utils import utils
label_desc: Use image node_id 1 to create a new SqueezeAxialPositionalEmbedding object from inherited base classes: nn with example: obj = SqueezeAxialPositionalEmbedding(dim, shape)