[英]How to switch axes (dimensions) in Julia for n-dimensional array
我有一个数组,我想切换轴的顺序。 它类似于转置,除了我想在尺寸大于2的数组上执行它。在Python中我会使用np.transpose和Matlab中的permute,但我似乎无法在Julia中找到它。 例如,
a = ones(2, 3, 4)
size(a)
(2,3,4)
从这里我想通过将轴(尺寸)重新排列为(2,3,1)来得到一个形状(3,4,2)的数组。 我正在寻找一个名为new_func的函数。
b = new_func(a, (2, 3, 1))
size(b)
(3,4,2)
根据Stefan Karpinski的说法,答案是Base.permutedims
( docs )。
例:
a = ones(2, 3, 4)
size(a) # => (2,3,4)
b = permutedims(a, [2, 3, 1])
size(b) # => (3,4,2)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.