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

Similar codes

0 vote267 views1 answer
Django many to many post update method via rest
Python
43 votes283 views6 answers
Df change column names
Python
0 vote258 views1 answer
Django update replace regex
Python
0 vote261 views2 answers
Avoid bad request django
Python
4 votes259 views1 answer
Round python with list
Python
-1 vote296 views1 answer
Fro flask import Flask, request, render_template, url_for, redirect, session
Python Flask
-1 vote269 views3 answers
Pd.read_excel column data type
Python
0 vote259 views1 answer
Python csv file tools
Python
1 vote284 views1 answer
How to make jupyterlab see other directory
Python
1 vote249 views1 answer
Qspinbox disable wheel python
Python