Torch.unsqueeze

Pytorch squeeze

x = torch.zeros(2, 1, 2, 1, 2)
x.size()
>>> torch.Size([2, 1, 2, 1, 2])

y = torch.squeeze(x) # remove 1
y.size()
>>> torch.Size([2, 2, 2])

y = torch.squeeze(x, 0)
y.size()
>>> torch.Size([2, 1, 2, 1, 2])

y = torch.squeeze(x, 1)
y.size()
>>> torch.Size([2, 2, 1, 2])

Source: pytorch.org

Torch.unsqueeze

a = torch.randn(4, 4, 4)
>>> torch.unsqueeze(a, 0).size()
torch.Size([1, 4, 4, 4])

>>> torch.unsqueeze(a, 1).size()
torch.Size([4, 1, 4, 4])

>>> torch.unsqueeze(a, 2).size()
torch.Size([4, 4, 1, 4])

>>> torch.unsqueeze(a, 3).size()
torch.Size([4, 4, 4, 1])

Source: linuxpip.org

Similar codes

0 vote117 views1 answer
Pygame get surface region
Python
1 vote126 views1 answer
How to normalize the data to get to the same range in python pandas
Python
5 votes131 views1 answer
How to print items in a list in a single line python
Python
0 vote139 views1 answer
Connectionrefusederror at /accounts/signup/ django allauth
Python
0 vote136 views1 answer
Dataframe passed by reference or value
Python
24 votes127 views6 answers
Pd.read_excel
Python
0 vote114 views1 answer
Discord.py how get user input
Python
6 votes121 views1 answer
Cv2 resize
Python
0 vote139 views1 answer
How to find pdf file in link beautifulsoup
Python
7 votes156 views3 answers
Update django
Python Django