Method data

deepspeed/runtime/utils.py:711–712

(self)

        return torch.LongTensor(data=meta).to(self.orig_device)

    def data(self):
        return self.local_data

    def local_size(self):
        return self.local_data.size()