oneflow.t

oneflow.t()

oneflow.t(input) → Tensor.

Expects input to be <= 2-D tensor and transposes dimensions 0 and 1.

0-D and 1-D tensors are returned as is. When input is a 2-D tensor this is equivalent to transpose(input, 0, 1).

Parameters

input (oneflow.Tensor) – An input tensor.

For example:

>>> import oneflow as flow
>>> import numpy as np

>>> x = flow.tensor(np.random.randn(), dtype=flow.float32)
>>> flow.t(x).shape
oneflow.Size([])
>>> x = flow.tensor(np.random.randn(3), dtype=flow.float32)
>>> flow.t(x).shape
oneflow.Size([3])
>>> x = flow.tensor(np.random.randn(2,3), dtype=flow.float32)
>>> flow.t(x).shape
oneflow.Size([3, 2])