def METHOD_NAME(self):
    embeds = SharedEmbedding.from_array(self.x, self.weight_path,
                                        'adagrad', 1.)
    indexs = paddle.to_tensor([0, 2])
    grad = [
        paddle.ones([2, 2]).astype('float32'),
        paddle.ones([2, ]).astype('float32')
    ]
    embeds.step_trace([indexs, grad])
    y = np.array([[-1., 2.], [5., 9.], [0., 6.]])
    self.assertTrue((embeds.weight == y).all())