torch.column_stack#

torch.column_stack(tensors,*,out=None)→Tensor#

Creates a new tensor by horizontally stacking the tensors intensors.

Equivalent totorch.hstack(tensors), except each zero or one dimensional tensortintensors is first reshaped into a(t.numel(),1) column before being stacked horizontally.

Parameters

tensors (sequence ofTensors) – sequence of tensors to concatenate

Keyword Arguments

out (Tensor,optional) – the output tensor.

Example:

>>>a=torch.tensor([1,2,3])>>>b=torch.tensor([4,5,6])>>>torch.column_stack((a,b))tensor([[1, 4],    [2, 5],    [3, 6]])>>>a=torch.arange(5)>>>b=torch.arange(10).reshape(5,2)>>>torch.column_stack((a,b,b))tensor([[0, 0, 1, 0, 1],        [1, 2, 3, 2, 3],        [2, 4, 5, 4, 5],        [3, 6, 7, 6, 7],        [4, 8, 9, 8, 9]])