Python mxnet.symbol.contrib.arange_like用法及代碼示例

用法:
mxnet.symbol.contrib.arange_like(data=None, start=_Null, step=_Null, repeat=_Null, ctx=_Null, axis=_Null, name=None, attr=None, out=None, **kwargs)

參數：

data：(Symbol) - 輸入
start：(double, optional, default=0) - 間隔開始。間隔包括此值。默認起始值為 0。
step：(double, optional, default=1) - 值之間的間距。
repeat：(int, optional, default='1') - 所有元素的重複時間。例如repeat=3，元素a將重複3次-> a, a, a。
ctx：(string, optional, default='') - 輸出上下文，格式為 [cpu|gpu|cpu_pinned](n)。僅用於命令式調用。
axis：(int or None, optional, default='None') - 根據輸入數組的某個軸的大小排列元素。負數被解釋為倒數。如果未提供，將根據輸入形狀排列元素。
name：(string, optional.) - 結果符號的名稱。

結果符號。

返回類型：

Symbol

返回具有均勻間隔值的數組。如果未給出軸，則輸出將具有與輸入數組相同的形狀。否則，輸出將是具有輸入形狀中指定軸大小的一維數組。

例子：

x = [[0.14883883 0.7772398  0.94865847 0.7225052 ]
     [0.23729339 0.6112595  0.66538996 0.5132841 ]
     [0.30822644 0.9912457  0.15502319 0.7043658 ]]
     <NDArray 3x4 @cpu(0)>

out = mx.nd.contrib.arange_like(x, start=0)

  [[ 0.  1.  2.  3.]
   [ 4.  5.  6.  7.]
   [ 8.  9. 10. 11.]]
   <NDArray 3x4 @cpu(0)>

out = mx.nd.contrib.arange_like(x, start=0, axis=-1)

  [0. 1. 2. 3.]
  <NDArray 4 @cpu(0)>

相關用法

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

用法:

參數：

返回：

返回類型：