tensorplay.functional.column_stack

tensorplay.functional.column_stack(tensors, *, out=None) → Tensor[source]

Creates a new tensor by horizontally stacking the tensors in tensors.

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

Parameters:

tensors (sequence of Tensors) – sequence of tensors to concatenate

Keyword Arguments:

out (Tensor, optional) – the output tensor.

Example:

>>> a = tensorplay.tensor([1, 2, 3])
>>> b = tensorplay.tensor([4, 5, 6])
>>> tensorplay.column_stack((a, b))
tensor([[1, 4],
    [2, 5],
    [3, 6]])
>>> a = tensorplay.arange(5)
>>> b = tensorplay.arange(10).reshape(5, 2)
>>> tensorplay.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]])