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


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


用法:

mxnet.ndarray.contrib.arange_like(data=None, start=_Null, step=_Null, repeat=_Null, ctx=_Null, axis=_Null, out=None, name=None, **kwargs)

參數

  • data(NDArray) - 輸入
  • 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') - 根據輸入數組的某個軸的大小排列元素。負數被解釋為倒數。如果未提供,將根據輸入形狀排列元素。
  • out(NDArray, optional) - 輸出 NDArray 來保存結果。

返回

out- 此函數的輸出。

返回類型

NDArray 或 NDArray 列表

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

例子:

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.ndarray.contrib.arange_like。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。