當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


Python mxnet.ndarray.random.randint用法及代碼示例

用法:

mxnet.ndarray.random.randint(low, high, shape=_Null, dtype=_Null, ctx=None, out=None, **kwargs)

參數

  • low(int, required) - 輸出間隔的下邊界。生成的所有值都將大於或等於低。
  • high(int, required) - 輸出區間的上邊界。生成的所有值都將小於高。
  • shape(int or tuple of ints, optional) - 要繪製的樣本數。如果形狀是,例如,(m, n)lowhigh是標量,輸出形狀將是(m, n).
  • dtype({'int32', 'int64'}, optional) - 輸出樣本的數據類型。默認為‘int32’
  • ctx(mxnet.context.Context, optional) - 輸出的設備上下文。默認為當前上下文。被覆蓋low.contextlow是一個 NDArray。
  • out(NDArray, optional) - 將輸出存儲到現有的 NDArray。

返回

dtype 類型的 NDArray。如果輸入 shape 具有形狀,例如 (m, n) ,則返回的 NDArray 的形狀將為 (m, n) 。返回的 NDArray 的內容將是來自區間 [low, high) 的樣本。

返回類型

ND陣列

從離散的均勻分布中抽取隨機樣本。

樣本在半開區間 [low, high) 上均勻分布(包括 low ,但不包括 high )。

例子

>>> mx.nd.random.randint(5, 100)
[ 90]
<NDArray 1 @cpu(0)
>>> mx.nd.random.randint(-10, 2, ctx=mx.gpu(0))
[ -8]
<NDArray 1 @gpu(0)>
>>> mx.nd.random.randint(-10, 10, shape=(2,))
[ -5  4]
<NDArray 2 @cpu(0)>

相關用法


注:本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.ndarray.random.randint。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。