triton.language.randint

triton.language.randint(seed, offset, n_rounds: constexpr = constexpr[10])

给定一个 seed 标量和一个 offset 块，返回一个包含随机 int32 的块。

如果你需要多个随机数流，使用 randint4x 可能比调用 randint 4 次更快。

参数:

• seed – 用于生成随机数的种子。

• offset – 要生成随机数的偏移量。