Python mxnet.ndarray.op.split用法及代碼示例

用法:
mxnet.ndarray.op.split(data=None, num_outputs=_Null, axis=_Null, squeeze_axis=_Null, out=None, name=None, **kwargs)

參數：

data：(NDArray) - 輸入
num_outputs：(int, required) - 分割數。請注意，這應該均勻地劃分axis.
axis：(int, optional, default='1') - 要分割的軸。
squeeze_axis：(boolean, optional, default=0) - 如果為 true，則從輸出數組的形狀中刪除長度為 1 的軸。注意那個設置squeeze_axis到true僅沿axis它是分裂的。還squeeze_axis可以設置為true除非input.shape[axis] == num_outputs.
out：(NDArray, optional) - 輸出 NDArray 來保存結果。

out：- 此函數的輸出。

返回類型：

NDArray 或 NDArray 列表

沿特定軸將數組拆分為多個子數組。

注意：

SliceChannel 已棄用。請改用split。

注意那num_outputs應該均勻地劃分分割數組的軸的長度。

例子：

x  = [[[ 1.]
       [ 2.]]
      [[ 3.]
       [ 4.]]
      [[ 5.]
       [ 6.]]]
x.shape = (3, 2, 1)

y = split(x, axis=1, num_outputs=2) // a list of 2 arrays with shape (3, 1, 1)
y = [[[ 1.]]
     [[ 3.]]
     [[ 5.]]]

    [[[ 2.]]
     [[ 4.]]
     [[ 6.]]]

y[0].shape = (3, 1, 1)

z = split(x, axis=0, num_outputs=3) // a list of 3 arrays with shape (1, 2, 1)
z = [[[ 1.]
      [ 2.]]]

    [[[ 3.]
      [ 4.]]]

    [[[ 5.]
      [ 6.]]]

z[0].shape = (1, 2, 1)

squeeze_axis=1從輸出數組的形狀中刪除長度為 1 的軸。注意那個設置squeeze_axis到1僅沿axis它是分裂的。還squeeze_axis隻有當input.shape[axis] == num_outputs.

例子：

z = split(x, axis=0, num_outputs=3, squeeze_axis=1) // a list of 3 arrays with shape (2, 1)
z = [[ 1.]
     [ 2.]]

    [[ 3.]
     [ 4.]]

    [[ 5.]
     [ 6.]]
z[0].shape = (2 ,1 )

相關用法

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

用法:

參數：

返回：

返回類型：

注意：