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


Python mxnet.symbol.random.shuffle用法及代碼示例


用法:

mxnet.symbol.random.shuffle(data, **kwargs)

參數

data(NDArray) - 輸入數據數組。

返回

表示輸入 data 的洗牌版本的新符號。

返回類型

Symbol

隨機打亂元素。

這會沿第一個軸對數組進行洗牌。每個子數組中元素的順序不會改變。例如,如果給定一個 2D 數組,則行的順序會隨機改變,但每行中元素的順序不會改變。

例子

>>> data = mx.nd.array([[0, 1, 2], [3, 4, 5], [6, 7, 8]])
>>> a = mx.sym.Variable('a')
>>> b = mx.sym.random.shuffle(a)
>>> b.eval(a=data)
[[ 0.  1.  2.]
 [ 6.  7.  8.]
 [ 3.  4.  5.]]
<NDArray 2x3 @cpu(0)>
>>> b.eval(a=data)
[[ 3.  4.  5.]
 [ 0.  1.  2.]
 [ 6.  7.  8.]]
<NDArray 2x3 @cpu(0)>

相關用法


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