'Create additional optimizer state.
For centered=False: n
For centered=True: n, g, delta
Parameters
weight : NDArray
The weight data'

def create_state(self, index, weight):

    if self.centered:
        return (mx.nd.zeros(weight.shape, weight.context), mx.nd.zeros(weight.shape, weight.context), mx.nd.zeros(weight.shape, weight.context))
    else:
        return (mx.nd.zeros(weight.shape, weight.context),)
