chainer.functions.swapaxes(x, axis1, axis2)[source]

指定した２軸のデータをスワップ(交換）する。

 

Parameters:
  • x (Variable or numpy.ndarray or cupy.ndarray) –入力値。 A (s1,s2,...,sN)(s1,s2,...,sN) -型の float 配列。
  • axis1 (int) – スワップする１つ目の軸
  • axis2 (int) – スワップする２つ目の軸
Returns:

軸をスワップしたVariable

Return type:

Variable

 

Example

  



>>> x = np.array([[[0, 1, 2], [3, 4, 5]]], 'f')

>>> x.shape
(1, 2, 3)

>>> y = F.swapaxes(x, axis1=0, axis2=1)

>>> y.shape
(2, 1, 3)

>>> y.data

array([[[ 0.,  1.,  2.]],

       [[ 3.,  4.,  5.]]], dtype=float32)