Np.column_stack equivalent

Hey guys,

is there np.column_stack equivalent in PyTorch? Or is a easy way to replicate its functionality by using stack, cat etc. Or should I just stick to np.column_stack (I don’t need to preserve gradient at the moment)

Thanks!

It is the same as stack(your_tensors, dim=0).

1 Like