mlx.data.Stream.repeat

mlx.data.Stream.repeat#

Stream.repeat(self: mlx.data._c.Stream, num_time: int) mlx.data._c.Stream#

Reset the stream num_time times before declaring it exhausted.

Parameters:

num_time (int) – How many times to repeat the dataset.