Creates a function that generates uniform random samples in [0, 1).
batch_size (int) – Number of batch samples
Tensor of shape (batch_size, dim_size) with uniform samples
torch.FloatTensor