Sample time steps from a specified distribution in [0, 1). Args: batch_size: Number of time samples to generate. time_sample_type: Type of distribution ('uniform', 'logitnormal', 'beta'). **kwargs: Additional parameters for non-uniform distributions.
(self, batch_size: int, time_sample_type: str = 'uniform', **kwargs)
source not stored for this graph (policy: none)