TensorFlow用expand_dim()来增加维度的方法

所需积分/C币:9 2020-12-24 10:35:01 38KB PDF
7
收藏 收藏
举报

TensorFlow中,想要维度增加一维,可以使用tf.expand_dims(input, dim, name=None)函数。当然,我们常用tf.reshape(input, shape=[])也可以达到相同效果,但是有些时候在构建图的过程中,placeholder没有被feed具体的值,这时就会包下面的错误:TypeError: Expected binary or unicode string, got 1 在这种情况下,我们就可以考虑使用expand_dims来将维度加1。比如我自己代码中遇到的情况,在对图像维度降到二维做特定操作后,要还原成四维[batch, height, wid

...展开详情
试读 1P TensorFlow用expand_dim()来增加维度的方法
立即下载 低至0.43元/次 身份认证VIP会员低至7折
一个资源只可评论一次,评论内容不能少于5个字
weixin_38578242 你的留言是对我莫大的支持
2020-12-24
您会向同学/朋友/同事推荐我们的CSDN下载吗?
谢谢参与!您的真实评价是我们改进的动力~
  • 至尊王者

    成功上传501个资源即可获取
关注 私信 TA的资源
上传资源赚积分or赚钱
    最新推荐
    TensorFlow用expand_dim()来增加维度的方法 9积分/C币 立即下载
    1/1
    TensorFlow用expand_dim()来增加维度的方法第1页

    试读结束, 可继续阅读

    9积分/C币 立即下载 >