def hybrid_forward(self, F, x, weight):
    return F.Embedding(x, weight, name='fwd', **self._kwargs)
def hybrid_forward(self, F, x, weight):
    return F.Embedding(x, weight, name='fwd', **self._kwargs)