Do you want to know the information regarding “tf.squeeze()”. If yes, you’re in the correct tutorial.
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]
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.