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

Python mxnet.ndarray.moveaxis用法及代码示例

用法:

mxnet.ndarray.moveaxis(tensor, source, destination)

参数：

  • tensor：(mx.nd.array) - 轴应该重新排序的数组
  • source：(int or sequence of int) - 要移动的轴的原始位置。可以是负数，但必须是唯一的。
  • destination：(int or sequence of int) - 每个原始轴的目标位置。可以是负数，但必须是唯一的。

返回：

result：- 带有移动轴的数组。

返回类型：

mx.nd.array

将 source 轴移动到 destination 位置，同时保持其他轴的原始顺序

例子：

>>> X = mx.nd.array([[1, 2, 3], [4, 5, 6]])
>>> mx.nd.moveaxis(X, 0, 1).shape
(3L, 2L)
>>> X = mx.nd.zeros((3, 4, 5))
>>> mx.nd.moveaxis(X, [0, 1], [-1, -2]).shape
(5, 4, 3)

相关用法


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