triton.language.randn

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

给定一个 seed 标量和一个 offset 块,返回一个服从 \(\mathcal{N}(0, 1)\) 分布的 float32 随机数块。

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

  • offsets – 用于生成随机数的偏移量。