Hi, i am moving from tensorflow to pytorch and i am looking for an equivalent to tf.data.Dataset.zip in pytorch.
Use case:
-
i have multiple quantities to precompute for a generated dataset dso (using sampling).
-
several datasets have to be created by mapping functions f1, f2, …fn: dsi = fi(dso) which can be scalars, vectors, matrices;
-
i want to be able to batch iterate all these datasets (including original one) simultaneously after an identical shuffle.
Is there an (elegant) to do this in pytorch ?
Thanks for any help.