in pai-python-sdk/training/pytorch_ddp/train_src/train_multinode.py [0:0]
def __init__(self, size): self.size = size self.data = [(torch.rand(20), torch.rand(1)) for _ in range(size)]