Mojo function
random_uniform
random_uniform[: origin.set, dtype: DType, rank: Int, //, output_fn: fn[Int, Int](idx: Index[$1], val: SIMD[dtype, $0]) capturing -> None, target: StringSlice[StaticConstantOrigin]](shape: Index[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: Index[$1], val: SIMD[dtype, $0]) capturing -> None
): The function which stores the generated values. - target (
StringSlice[StaticConstantOrigin]
): The target to run on.
Args:
- shape (
Index[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.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!