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