/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

7 votes314 views4 answers
Pytorch save model
Python
30 votes323 views4 answers
How to round a number in python
Python
1 vote311 views1 answer
How shorten with enter long script python
Python
3 votes347 views2 answers
Draw bounding box on image python opencv
Python
18 votes780 views4 answers
Py to exe converter online
Python
0 vote324 views1 answer
Pandas .replace multiple values in column
Python
1 vote336 views1 answer
How to add decorators with class in django
Python
0 vote322 views1 answer
Python bubble plot
Python
0 vote306 views1 answer
Python - Comment Parse String to List
Python
0 vote347 views1 answer
Load image metadata with pil
Python