chainerx.transpose

chainerx.transpose(a, axes=None)

Permutes the dimensions of an array.

Parameters
  • a (ndarray) – Array to permute the dimensions.

  • axes (tuple of ints) – Permutation of the dimensions. This function reverses the shape by default.

Returns

A view of a with the dimensions permuted.

Return type

ndarray

Note

During backpropagation, this function propagates the gradient of the output array to the input array a.