tf.squeeze
- z = tf.squeeze(y) #删除y中所有为1的维度;
- z = tf.squeeze(y,[3,5]) #删除[3,5]中所有为1的维度;
tf.expand_dims
- tf中没有tf.unsqueeze,一般写tf.expand_dims()扩增维度;
- one_img = tf.expand_dims(one_img, 0) #第0维插入1维
- one_img = tf.expand_dims(one_img, -1) #-1表示最后一维
tensorflow中squeeze与expend_dims
阅读 26
2022-01-08
相关推荐
精彩评论(0)