jax.random.ball

jax.random.ball(key, d, p=2, shape=(), dtype=<class 'float'>)

從單位 Lp 球均勻取樣。

參考文獻：https://arxiv.org/abs/math/0503650。

參數：

• key (ArrayLike) – 作為隨機金鑰使用的 PRNG 金鑰。

• d (int) – 代表球體維度的非負整數。

• p (float) – 代表 Lp 範數的 p 參數的浮點數。

• shape (Shape) – 選填，結果的批次維度。預設為 ()。

• dtype (DTypeLikeFloat) – 選填，傳回值的浮點數資料型別 (如果 jax_enable_x64 為 true，則預設為 float64，否則為 float32)。

傳回：

形狀為 (*shape, d) 且具有指定資料型別的隨機陣列。