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


Python mxnet.ndarray.op.repeat用法及代码示例


用法:

mxnet.ndarray.op.repeat(data=None, repeats=_Null, axis=_Null, out=None, name=None, **kwargs)

参数

  • data(NDArray) - 输入数据数组
  • repeats(int, required) - 每个元素的重复次数。
  • axis(int or None, optional, default='None') - 沿其重复值的轴。负数被解释为倒数。默认情况下,使用扁平化的输入数组,并返回一个扁平的输出数组。
  • out(NDArray, optional) - 输出 NDArray 来保存结果。

返回

out- 此函数的输出。

返回类型

NDArray 或 NDArray 列表

重复数组的元素。默认情况下,repeat 将输入数组展平为一维,然后重复元素:

x = [[ 1, 2],
     [ 3, 4]]
repeat(x, repeats=2) = [ 1.,  1.,  2.,  2.,  3.,  3.,  4.,  4.]
参数 axis 指定执行重复的轴:
重复(x,重复=2,轴=1)= [[ 1., 1., 2., 2.],

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

重复(x,重复=2,轴=0)= [[ 1., 2.],

[1., 2.], [3., 4.], [3., 4.]]

重复(x,重复=2,轴=-1)= [[ 1., 1., 2., 2.],

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

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.op.repeat。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。