/Python/RuntimeError: input must have 3 dimensions, got 4 site:stackoverflow.com

RuntimeError: input must have 3 dimensions, got 4 site:stackoverflow.com

import torch
r = torch.tensor([[[5,10,11,18,19],[900,98,97,51,64],[89,87,45,64,88]]])#this is a 4 dimensions
print(r)
>>> tensor([[[5,10,11,18,19],[900,98,97,51,64],[89,87,45,64,88]]])
#if we convert it into 3 dimensions from 4 dimensions,we need to use torch.squeeze() function
r.squeeze()
>>> tensor([[5,10,11,18,19],[900,98,97,51,64],[89,87,45,64,88]])

Similar codes

0 vote140 views1 answer
Plt add y gridlines
Python
2 votes142 views1 answer
Distribution seaborn
Python
20 votes123 views3 answers
Accommodation
Python
0 vote136 views1 answer
Python dash bootstrap buttons with icons
Python
3 votes92 views1 answer
Torch mse loss
Python
2 votes183 views2 answers
Django template array index
Python Django
0 vote138 views1 answer
Where are spacy models stored
Python
2 votes136 views1 answer
Swapcase
Python
2 votes134 views1 answer
Termcolor python
Python
0 vote132 views2 answers
Remove unicode from string python
Python