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])