当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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