首页
学习
活动
专区
工具
TVP
发布

tf.pad

作者头像
狼啸风云
修改2022-09-04 20:52:51
1.2K0
修改2022-09-04 20:52:51
举报
tf.pad(
    tensor,
    paddings,
    mode='CONSTANT',
    name=None,
    constant_values=0
)

pad一个张量。 这个操作根据指定的paddings填充一个tensor。padding是一个形状为[n, 2]的整数张量,其中n是张量的秩。对于输入的每个维度D,paddings[D, 0]表示在该维度tensor的内容之前要添加多少个值,paddings[D, 1]表示在该维度张量的内容之后要添加多少个值。如果mode是“REFLECT”,那么paddings[D, 0]和paddins[D, 1]必须不大于tensor.dim_size(D) - 1。如果mode是“SYMMETRIC”,那么paddings[D, 0]和paddings[D, 1]必须不大于tensor.dim_size(D)。

输出各维度D的padding尺寸为:

paddings[D, 0] + tensor.dim_size(D) + paddings[D, 1]
paddings[D, 0] + tensor.dim_size(D) + paddings[D, 1]

例如:

t = tf.constant([[1, 2, 3], [4, 5, 6]])
paddings = tf.constant([[1, 1,], [2, 2]])
# 'constant_values' is 0.
# rank of 't' is 2.
tf.pad(t, paddings, "CONSTANT")  # [[0, 0, 0, 0, 0, 0, 0],
                                 #  [0, 0, 1, 2, 3, 0, 0],
                                 #  [0, 0, 4, 5, 6, 0, 0],
                                 #  [0, 0, 0, 0, 0, 0, 0]]

tf.pad(t, paddings, "REFLECT")  # [[6, 5, 4, 5, 6, 5, 4],
                                #  [3, 2, 1, 2, 3, 2, 1],
                                #  [6, 5, 4, 5, 6, 5, 4],
                                #  [3, 2, 1, 2, 3, 2, 1]]

tf.pad(t, paddings, "SYMMETRIC")  # [[2, 1, 1, 2, 3, 3, 2],
                                  #  [2, 1, 1, 2, 3, 3, 2],
                                  #  [5, 4, 4, 5, 6, 6, 5],
                                  #  [5, 4, 4, 5, 6, 6, 5]]

参数:

  • tensor: 一个张量。
  • paddings: 一个int32类型的张量
  • mode "CONSTANT"、 "REFLECT" 或 "SYMMETRIC" (不区分大小写)之一。
  • name:操作的名称(可选)。
  • constant_values: 在“CONSTANT”模式下,要使用标量填充值。必须和tensor的类型相同。

返回值:

  • 一个张量。与参数张量的类型相同。

可能产生的异常:

  • ValueError: When mode is not one of "CONSTANT", "REFLECT", or "SYMMETRIC".

原链接: https://tensorflow.google.cn/versions/r1.9/api_docs/python/tf/pad?hl=en

本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2019年06月24日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档