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


Python mxnet.symbol.argmin用法及代碼示例


用法:

mxnet.symbol.argmin(data=None, axis=_Null, keepdims=_Null, name=None, attr=None, out=None, **kwargs)

參數

  • data(Symbol) - 輸入
  • axis(int or None, optional, default='None') - 執行縮減的軸。負值表示從右到左索引。Requires axis to be set as int, because global reduction is not supported yet.
  • keepdims(boolean, optional, default=0) - 如果設置為True,縮小的軸作為尺寸為一的維度留在結果中。
  • name(string, optional.) - 結果符號的名稱。

返回

結果符號。

返回類型

Symbol

返回沿軸的最小值的索引。

在多次出現最小值的情況下,返回與第一次出現對應的索引。

例子:

x = [[ 0.,  1.,  2.],
     [ 3.,  4.,  5.]]

// argmin along axis 0
argmin(x, axis=0) = [ 0.,  0.,  0.]

// argmin along axis 1
argmin(x, axis=1) = [ 0.,  0.]

// argmin along axis 1 keeping same dims as an input array
argmin(x, axis=1, keepdims=True) = [[ 0.],
                                    [ 0.]]

相關用法


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