https://pytorch-datastream.readthedocs.io/en/latest/datastream.html#datastream.Datastream.multi_sample
