oneflow.atleast_3d

oneflow.atleast_3d(*tensors) → Tensor or List[Tensor]

Returns a 3-dimensional view of each input tensor with zero dimensions. Input tensors with three or more dimensions are returned as-is.

The interface is consistent with PyTorch.

The documentation is referenced from: https://pytorch.org/docs/1.10/generated/torch.atleast_3d.html.

Parameters

tensors (List[oneflow.Tensor] or oneflow.Tensor) – Tensor or list of tensors to be reshaped

Returns

A Tensor

For example:

>>> import oneflow as flow
>>> x = flow.tensor(0)
>>> flow.atleast_3d(x).shape
oneflow.Size([1, 1, 1])
>>> x = flow.randn(3)
>>> flow.atleast_3d(x).shape
oneflow.Size([1, 3, 1])
>>> x = flow.randn(3, 4)
>>> flow.atleast_3d(x).shape
oneflow.Size([3, 4, 1])
>>> x = flow.randn(3, 4, 5)
>>> flow.atleast_3d(x).shape
oneflow.Size([3, 4, 5])