當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。