tf.squeeze()

Do you want to know the information regarding “tf.squeeze()”. If yes, you’re in the correct tutorial.

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]

Final Thoughts

I hope this article helps you to know about “tf.squeeze()”. If you have any doubts regarding this post please let us know via the comment section. Share this tutorial with your friends and family via social networks.

Hi, I'm Ranjith a full-time Blogger, YouTuber, Affiliate Marketer, & founder of Coder Diksha. Here, I post about programming to help developers.

Share on:

Leave a Comment