/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

2 votes344 views1 answer
W=how to tell if decimal in python
Python
3 votes344 views3 answers
How to take input in 2d list in python
Python
0 vote360 views1 answer
MASS algorithm python
Python
0 vote376 views1 answer
Python cv2 how to update image
Python
0 vote401 views1 answer
Django 3.2 compatible to python 3.10?
Python Django
0 vote341 views1 answer
How to get minimum value in list python
Python
1 vote359 views1 answer
Elif in django template
Python
0 vote337 views2 answers
Plt.hist bins
Python
0 vote375 views3 answers
Brew PIP
Python
10 votes349 views6 answers
Formatting in python
Python