code: def forward(self, x_float):
      x_embedding = 0
      for i in range(x_float.shape[1]):
          x = x_float[:, i]
          x = paddle.reshape(x, [-1, 1])
          if i == 0:
              gaussian_expansion = paddle.exp(
                  -((x - self.centers1) ** 2) / self.width**2
              )
          elif i == 1:
              gaussian_expansion = paddle.exp(
                  -((x - self.centers2) ** 2) / self.width**2
              )
  
          x_embedding += self.atom_embedding_list[i](gaussian_expansion)
      return x_embedding
  
imports: import numpy
  import paddle
  import paddle.nn
  import paddle.distributed
  from pgl.utils.logger import log
  import features.local_feature
label_desc: Use image node_id 2 for calling the AtomEncoderFloat obj's underlying member method code with example usage: obj.forward(x_float) and returns: x_embedding