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!
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)
.