What’s the difference between torch.stack() and torch.cat() functions?

by

stack

Concatenates sequence of tensors along a new dimension.

cat

Concatenates the given sequence of seq tensors in the given dimension.

So if A and B are of shape (3, 4):

  • torch.cat([A, B], dim=0) will be of shape (6, 4)
  • torch.stack([A, B], dim=0) will be of shape (2, 3, 4)

More Related Contents:

  1. How to initialize weights in PyTorch?
  2. How do I initialize weights in PyTorch?
  3. How does Pytorch’s “Fold” and “Unfold” work?
  4. What does model.eval() do in pytorch?
  5. RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
  6. PyTorch Binary Classification – same network structure, ‘simpler’ data, but worse performance?
  7. Will it lead to Overfitting / Curse of Dimensionality
  8. How to get the location of all text present in an image using OpenCV?
  9. Best way to save a trained model in PyTorch? [closed]
  10. What is a `”Python”` layer in caffe?
  11. How to apply gradient clipping in TensorFlow?
  12. How do I save a trained model in PyTorch?
  13. Keras Sequential model input layer
  14. Why do we need to call zero_grad() in PyTorch?
  15. Does Any one got “AttributeError: ‘str’ object has no attribute ‘decode’ ” , while Loading a Keras Saved Model
  16. Keras custom loss function: Accessing current input pattern
  17. PyTorch NotImplementedError in forward
  18. How can I use a pre-trained neural network with grayscale images?
  19. How to understand the term `tensor` in TensorFlow?
  20. Pytorch – RuntimeError: Trying to backward through the graph a second time, but the buffers have already been freed
  21. Removing then Inserting a New Middle Layer in a Keras Model
  22. Unbalanced data and weighted cross entropy
  23. What is the role of TimeDistributed layer in Keras?
  24. How to handle non-determinism when training on a GPU?
  25. keras: how to save the training history attribute of the history object
  26. How to construct a network with two inputs in PyTorch
  27. How to fix “ResourceExhaustedError: OOM when allocating tensor”
  28. How do I split a custom dataset into training and test datasets?
  29. RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! when resuming training
  30. Pytorch ValueError: optimizer got an empty parameter list

Leave a Comment