triton.language.randint

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

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

如果需要生成多组随机数, 使用randint4x可能比调用4次randint更快。

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

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