def to_kwargs(self, inputs, kwargs, device_id):
    return scatter_kwargs(inputs, kwargs, [device_id], dim=self.dim)
def to_kwargs(self, inputs, kwargs, device_id):
    return scatter_kwargs(inputs, kwargs, [device_id], dim=self.dim)