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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法:

参数：

返回：

返回类型：

注意：