medium10 pts · first answer countsdeep-learningpytorchtensors
torch.cat or torch.stack
a and b both have shape (4, 8). What are the shapes of torch.cat([a, b], dim=0) and torch.stack([a, b], dim=0)?
Discussion 0
Join free to post in the discussion.
Related questions
- Dropout after eval()Machine Learning Engineer
- Parameters in nn.Linear(10, 5)Machine Learning Engineer
- Why sigmoid slows deep trainingMachine Learning Engineer
- Range of tanhMachine Learning Engineer
- Cross-entropy loss by handMachine Learning Engineer
- Telling autograd to track a tensorMachine Learning Engineer
- Why wrap inference in no_gradMachine Learning Engineer
- Logging loss without leaking memoryMachine Learning Engineer