Mojo function

random_uniform

random_uniform[: origin.set, dtype: DType, rank: Int, //, output_fn: fn[Int, Int](idx: IndexList[$1], val: SIMD[dtype, $0]) capturing -> None, target: StringSlice[StaticConstantOrigin]](shape: IndexList[rank], lower_bound: SIMD[dtype, 1], upper_bound: SIMD[dtype, 1], seed_value: SIMD[uint64, 1], ctx: DeviceContextPtr)

Call output_fn with values generated from a uniform distribution on [lower_bound, upper_bound] for floating-point types or [lower_bound, upper_bound) for integer types.

Parameters:

dtype (DType): The data type to generate.
rank (Int): The rank of the underlying buffer.
output_fn (fn[Int, Int](idx: IndexList[$1], val: SIMD[dtype, $0]) capturing -> None): The function which stores the generated values.
target (StringSlice[StaticConstantOrigin]): The target to run on.

Args:

shape (IndexList[rank]): The shape of the output being stored into by output_fn.
lower_bound (SIMD[dtype, 1]): The lower bound on the uniform range.
upper_bound (SIMD[dtype, 1]): The upper bound on the uniform range.
seed_value (SIMD[uint64, 1]): Seed value used to initialize the random number generator.
ctx (DeviceContextPtr): The device context.