PyTorch .row_stack()

In PyTorch, the .row_stack() function stacks or arranges a sequence of tensors vertically (row-wise). It is an alias or alternative for the .vstack() function.

Syntax

torch.row_stack(tensors, *, out=None)

• tensors: The sequence of tensors to be stacked vertically.
• out (Optional): A tensor to store the output. It must have the correct shape to accommodate the result.

Example

The following example demonstrates the usage of the .row_stack() function:

import torch

# Create two tensors
ten1 = torch.tensor([12, 23, 34])
ten2 = torch.tensor([45, 56, 67])

# Stack the tensors vertically
res = torch.row_stack((ten1, ten2))

# Print the resultant tensor
print(res)

Copy to clipboard
Copy to clipboard

The above code produces the following output:

tensor([[12, 23, 34],
        [45, 56, 67]])

Copy to clipboard
Copy to clipboard