oneflow.swapaxes

oneflow.swapaxes(input, axis0, axis1)Tensor

This function is equivalent to NumPy’s swapaxes function.

For example:

>>> import oneflow as flow

>>> x = flow.tensor([[[0,1],[2,3]],[[4,5],[6,7]]])
>>> x.shape
oneflow.Size([2, 2, 2])
>>> flow.swapaxes(x, 0, 1).shape
oneflow.Size([2, 2, 2])
>>> flow.swapaxes(x, 0, 2).shape
oneflow.Size([2, 2, 2])