hstack(tensors) → Tensor¶
Stack tensors in
tensorshorizontally (column wise).
This is equivalent to concatenation tensors in
tensorsalong the first axis for 1-D tensors, and along the second axis for all other tensors.
When there are tensors with dimension less than 1, these tensors will be reshaped by
oneflow.atleast_1d()to 1-dims tensors before stacking.
The documentation is referenced from: https://pytorch.org/docs/1.10/generated/torch.hstack.html.
tensors – (List[oneflow.Tensor]): sequence of tensors to stack
>>> import oneflow as flow >>> x1 = flow.randn(5, 2) >>> x2 = flow.randn(5, 3) >>> flow.hstack([x1, x2]).shape oneflow.Size([5, 5]) >>> x = flow.randn(5) >>> flow.hstack([x, x]).shape oneflow.Size()