tf.expand_dims

最新推荐文章于 2021-11-28 19:32:24 发布

cc和ss

最新推荐文章于 2021-11-28 19:32:24 发布

阅读量146

点赞数

文章标签：机器学习 tensorflow

本文链接：https://blog.youkuaiyun.com/qq_36521174/article/details/116796016

版权

加一个维度，

ne_img = tf.expand_dims(one_img, -1) #-1表示最后一维

官方的例子和说明

# 't' is a tensor of shape [2]
shape(expand_dims(t, 0)) ==> [1, 2]
shape(expand_dims(t, 1)) ==> [2, 1]
shape(expand_dims(t, -1)) ==> [2, 1]

# 't2' is a tensor of shape [2, 3, 5]
shape(expand_dims(t2, 0)) ==> [1, 2, 3, 5]
shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5]
shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1]