Tf.squeeze()

Removes dimensions of size 1 from the shape of a tensor.

tf.squeeze(
    input, axis=None, name=None
)

Given a tensor input, this operation returns a tensor of the same 
type with all dimensions of size 1 removed. If you don't want to 
remove all size 1 dimensions, you can remove specific size 1 
dimensions by specifying axis.

Example:
# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
tf.shape(tf.squeeze(t))  # [2, 3]
# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
tf.shape(tf.squeeze(t, [2, 4]))  # [1, 2, 3, 1]

Source: tensorflow.org

Similar codes

0 vote109 views1 answer
From .cv2 import * ImportError: /home/pi/.local/lib/python3.7/site-packages/cv2/cv2.cpython-37m-arm-linux-gnueabihf.so: undefined symbol: __atomic_fetch_add_8
Python
1 vote111 views2 answers
How to make a game in python
Python
0 vote100 views2 answers
Removing rows dataframe not in another dataframe using two columns
Python
9 votes116 views4 answers
Capitalize first letter of each word python
Python
1 vote101 views1 answer
Python nested tqdm
Python
1 vote139 views1 answer
Display category field in django admin
Python Django
0 vote101 views2 answers
Arg parse array argument
Python
0 vote110 views1 answer
DatetimeProperties' object has no attribute 'weekday_name'
Python
2 votes102 views1 answer
Create Pandas Data Frame in Python
Python
2 votes119 views3 answers
Python split string size
Python